From c76608ee4e7b6c500261dcf8b6ea7af378553416 Mon Sep 17 00:00:00 2001 From: conorheins Date: Thu, 8 Dec 2022 13:15:23 +0100 Subject: [PATCH] Import new environments in __init__.py file for envs library --- pymdp/envs/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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