Have autoreload update shell namespace with reloaded module variables (…

…#2278) * Have autoreload update shell namespace with reloaded module variables * Update comments
3b1b · Dec 13, 2024 · 39fbb67 · 39fbb67
1 parent c13d2a9
commit 39fbb67
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/docs/source/getting_started/configuration.rst b/docs/source/getting_started/configuration.rst
@@ -63,7 +63,7 @@ flag                                                       abbr   function
 ``--video_dir VIDEO_DIR``                                         Directory to write video
 ``--config_file CONFIG_FILE``                                     Path to the custom configuration file
 ``--log-level LOG_LEVEL``                                         Level of messages to Display, can be DEBUG / INFO / WARNING / ERROR / CRITICAL
-``--autoreload``                                                  Automatically reload Python modules to pick up code changes across different files
+``--autoreload``                                                  Automatically reload Python modules to pick up code changes across during an interactive embedding
 ========================================================== ====== =====================================================================================================================================================================================================
 
 custom_config

diff --git a/manimlib/scene/scene_embed.py b/manimlib/scene/scene_embed.py
@@ -142,9 +142,12 @@ def reload_scene(self, embed_line: int | None = None) -> None:
         self.shell.run_line_magic("exit_raise", "")
 
     def auto_reload(self):
-        """Enables IPython autoreload for automatic reloading of modules."""
-        self.shell.magic("load_ext autoreload")
-        self.shell.magic("autoreload all")
+        """Enables reload the shell's module before all calls"""
+        def pre_cell_func(*args, **kwargs):
+            new_mod = ModuleLoader.get_module(self.shell.user_module.__file__, is_during_reload=True)
+            self.shell.user_ns.update(vars(new_mod))
+
+        self.shell.events.register("pre_run_cell", pre_cell_func)
 
     def checkpoint_paste(
         self,