From 3cee3de94f2872a5b700d8b27b227f35e13a9cc7 Mon Sep 17 00:00:00 2001 From: zavden <51290473+zavden@users.noreply.github.com> Date: Wed, 24 Jul 2019 18:01:12 -0500 Subject: [PATCH] Update extract_scene.py --- manimlib/extract_scene.py | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) diff --git a/manimlib/extract_scene.py b/manimlib/extract_scene.py index 0cf78363..ed486152 100644 --- a/manimlib/extract_scene.py +++ b/manimlib/extract_scene.py @@ -64,6 +64,8 @@ def is_child_scene(obj, module): return False if obj == Scene: return False + if not obj.__module__.startswith(module.__name__): + return False return True @@ -118,13 +120,16 @@ def get_scenes_to_render(scene_classes, config): def get_scene_classes_from_module(module): - return [ - member[1] - for member in inspect.getmembers( - module, - lambda x: is_child_scene(x, module) - ) - ] + if hasattr(module, "SCENES_IN_ORDER"): + return module.SCENES_IN_ORDER + else: + return [ + member[1] + for member in inspect.getmembers( + module, + lambda x: is_child_scene(x, module) + ) + ] def main(config):