Simplify enqueueing

dduugg · Dec 2, 2023 · 8ea65e3 · 8ea65e3
1 parent e2650f1
commit 8ea65e3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/yard-sorbet/node_utils.rb b/lib/yard-sorbet/node_utils.rb
@@ -30,9 +30,9 @@ def self.bfs_traverse(node, &_blk)
     # Enqueue the eligible children of a node in the BFS queue
     sig { params(queue: Queue, node: YARD::Parser::Ruby::AstNode).void }
     def self.enque_children(queue, node)
-      num_children = node.children.size
-      node.children.each_with_index do |child, i|
-        next if i == num_children - 1 &&
+      last_child = node.children.last
+      node.children.each do |child|
+        next if child == last_child &&
                 node.is_a?(YARD::Parser::Ruby::MethodCallNode) &&
                 SKIP_METHOD_CONTENTS.include?(node.method_name(true))