Import new environments in __init__.py file for envs library

infer-actively · Dec 8, 2022 · c76608e · c76608e
1 parent 9c05445
commit c76608e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymdp/envs/__init__.py b/pymdp/envs/__init__.py
@@ -1,4 +1,4 @@
 from .env import Env
 from .grid_worlds import GridWorldEnv, DGridWorldEnv
-from .visual_foraging import VisualForagingEnv
-from .tmaze import TMazeEnv, TMazeEnvNullOutcome
+from .visual_foraging import VisualForagingEnv, SceneConstruction, RandomDotMotion, initialize_scene_construction_GM, initialize_RDM_GM
+from .tmaze import TMazeEnv, TMazeEnvNullOutcome