Update chunk_python_code.py
Browse files- 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,
|
58 |
elif isinstance(first_level_node, ast.FunctionDef):
|
59 |
documents.extend(
|
60 |
-
_chunk_first_level_func_node(first_level_node,
|
61 |
elif isinstance(first_level_node, ast.Assign):
|
62 |
documents.extend(
|
63 |
-
_chunk_first_level_assign_node(first_level_node,
|
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))
|