opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-07-30 21:44:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix 'Failed to get module" warning with Scene embed

Browse Source
This commit is contained in:
Grant Sanderson
2022-12-18 11:11:08 -08:00
parent e728196814
commit af471161ea
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
manimlib/scene/scene.py
View File

@ -1,7 +1,6 @@
from __future__ import annotations from __future__ import annotations
from collections import OrderedDict from collections import OrderedDict
from functools import wraps
import inspect import inspect
import os import os
import platform import platform
@ -14,6 +13,7 @@ from tqdm import tqdm as ProgressDisplay
from manimlib.animation.animation import prepare_animation from manimlib.animation.animation import prepare_animation
from manimlib.camera.camera import Camera from manimlib.camera.camera import Camera
from manimlib.config import get_module
from manimlib.constants import ARROW_SYMBOLS from manimlib.constants import ARROW_SYMBOLS
from manimlib.constants import DEFAULT_WAIT_TIME from manimlib.constants import DEFAULT_WAIT_TIME
from manimlib.constants import COMMAND_MODIFIER from manimlib.constants import COMMAND_MODIFIER
@ -164,11 +164,11 @@ class Scene(object):
# the hood calling the pyglet event loop # the hood calling the pyglet event loop
if self.window is None: if self.window is None:
return return
log.info( log.info("""
"Tips: You are now in the interactive mode. Now you can use the keyboard" Tips: You are now in the interactive mode. Now you can use the keyboard
" and the mouse to interact with the scene. Just press `command + q` or `esc`" and the mouse to interact with the scene. Just press `command + q` or `esc`
" if you want to quit." if you want to quit.
) """)
self.skip_animations = False self.skip_animations = False
self.refresh_static_mobjects() self.refresh_static_mobjects()
while not self.is_window_closing(): while not self.is_window_closing():
@ -187,7 +187,9 @@ class Scene(object):
shell = embed.InteractiveShellEmbed.instance() shell = embed.InteractiveShellEmbed.instance()
# Use the locals namespace of the caller # Use the locals namespace of the caller
local_ns = inspect.currentframe().f_back.f_locals caller_frame = inspect.currentframe().f_back
local_ns = dict(caller_frame.f_locals)
# Add a few custom shortcuts # Add a few custom shortcuts
local_ns.update({ local_ns.update({
name: getattr(self, name) name: getattr(self, name)
@ -197,6 +199,8 @@ class Scene(object):
] ]
}) })
module = get_module(caller_frame.f_globals["__file__"])
# This is useful if one wants to re-run a block of scene # This is useful if one wants to re-run a block of scene
# code, while developing, tweaking it each time. # code, while developing, tweaking it each time.
# As long as the copied selection starts with a comment, # As long as the copied selection starts with a comment,
@ -246,7 +250,7 @@ class Scene(object):
shell.events.register("post_run_cell", post_cell_func) shell.events.register("post_run_cell", post_cell_func)
shell(local_ns=local_ns, stack_depth=2) shell(local_ns=local_ns, stack_depth=2, module=module)
# End scene when exiting an embed # End scene when exiting an embed
if close_scene_on_exit: if close_scene_on_exit: