bupa1018 commited on
Commit
9a5dce7
·
1 Parent(s): 64756f0

Update chunk_python_code.py

Browse files
Files changed (1) hide show
  1. chunk_python_code.py +3 -3
chunk_python_code.py CHANGED
@@ -54,13 +54,13 @@ def _iterate_ast(python_code, documents, file_path):
54
 
55
  if isinstance(first_level_node, ast.ClassDef):
56
  documents.extend(
57
- _handle_first_level_class(first_level_node, documents, python_code))
58
  elif isinstance(first_level_node, ast.FunctionDef):
59
  documents.extend(
60
- _chunk_first_level_func_node(first_level_node, documents, python_code))
61
  elif isinstance(first_level_node, ast.Assign):
62
  documents.extend(
63
- _chunk_first_level_assign_node(first_level_node, documents, python_code))
64
  else:
65
  documents.extend(
66
  _handle_not_defined_case(python_code))
 
54
 
55
  if isinstance(first_level_node, ast.ClassDef):
56
  documents.extend(
57
+ _handle_first_level_class(first_level_node, python_code))
58
  elif isinstance(first_level_node, ast.FunctionDef):
59
  documents.extend(
60
+ _chunk_first_level_func_node(first_level_node, python_code))
61
  elif isinstance(first_level_node, ast.Assign):
62
  documents.extend(
63
+ _chunk_first_level_assign_node(first_level_node, python_code))
64
  else:
65
  documents.extend(
66
  _handle_not_defined_case(python_code))