Spaces:
Sleeping
Sleeping
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))
|