diff --git a/pymdp/envs/__init__.py b/pymdp/envs/__init__.py index 5b70268c..e461e569 100644 --- 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 \ No newline at end of file +from .visual_foraging import VisualForagingEnv, SceneConstruction, RandomDotMotion, initialize_scene_construction_GM, initialize_RDM_GM +from .tmaze import TMazeEnv, TMazeEnvNullOutcome