Update RunCodeFileEditAtomicFlow.py
Browse files
RunCodeFileEditAtomicFlow.py
CHANGED
@@ -59,7 +59,7 @@ class RunCodeFileEditAtomicFlow(CodeFileEditAtomicFlow):
|
|
59 |
# to re-import the library.
|
60 |
code_lang = input_data["language"].lower()
|
61 |
code_lib_name = os.path.splitext(os.path.basename(code_lib_location))[0]
|
62 |
-
if code_lang == 'python' and code_lib_name in code_str:
|
63 |
prepend_code = f"import importlib\nimport {code_lib_name}\nimportlib.reload({code_lib_name})\n"
|
64 |
code_str = prepend_code + code_str
|
65 |
|
|
|
59 |
# to re-import the library.
|
60 |
code_lang = input_data["language"].lower()
|
61 |
code_lib_name = os.path.splitext(os.path.basename(code_lib_location))[0]
|
62 |
+
if code_lang == 'python' and code_lib_name in code_str and 'importlib.reload' not in code_str:
|
63 |
prepend_code = f"import importlib\nimport {code_lib_name}\nimportlib.reload({code_lib_name})\n"
|
64 |
code_str = prepend_code + code_str
|
65 |
|