code: def get_comp_fors(comp_for):
      yield comp_for
      last = comp_for.children[-1]
      while True:
          if last.type == "comp_for":
              yield last
          elif not last.type == "comp_if":
              break
          last = last.children[-1]
  
file_path: catboost/contrib/python/jedi/py2/jedi/parser_utils.py
imports: import re
  import textwrap
  from inspect import cleandoc
  from parso.python import tree
  from parso.cache import parser_cache
  from jedi._compatibility import literal_eval, force_unicode
label_desc: Use image node_id 2 for calling a global function with example usage: get_comp_fors(comp_for) without return types