diff --git a/great_ai/__main__.py b/great_ai/__main__.py index 481c802..98396ba 100644 --- a/great_ai/__main__.py +++ b/great_ai/__main__.py @@ -139,6 +139,8 @@ def get_script_name(file_name_argument: str) -> str: exporter = PythonExporter() content, _ = exporter.from_filename(file_name_argument) + content = re.sub(r".*get_ipython\(.*", "", content) + file_name_argument = get_script_name_of_notebook(file_name_argument) with open(file_name_argument, "w", encoding="utf-8") as f: