From ce34df181458cbd11e87dde707b946794ec87c44 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 23 Jul 2023 11:05:45 +0200 Subject: [PATCH] No longer export old custom make --- fancy_gym/__init__.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/fancy_gym/__init__.py b/fancy_gym/__init__.py index f6f690a..32308fa 100644 --- a/fancy_gym/__init__.py +++ b/fancy_gym/__init__.py @@ -1,5 +1,6 @@ from fancy_gym import dmc, meta, open_ai -from fancy_gym.utils.make_env_helpers import make, make_bb, make_rank +from fancy_gym.utils.make_env_helpers import make_bb +from .envs.registry import register from .dmc import ALL_DMC_MOVEMENT_PRIMITIVE_ENVIRONMENTS # Convenience function for all MP environments from .envs import ALL_FANCY_MOVEMENT_PRIMITIVE_ENVIRONMENTS @@ -7,7 +8,5 @@ from .meta import ALL_METAWORLD_MOVEMENT_PRIMITIVE_ENVIRONMENTS from .open_ai import ALL_GYM_MOVEMENT_PRIMITIVE_ENVIRONMENTS ALL_MOVEMENT_PRIMITIVE_ENVIRONMENTS = { - key: value + ALL_DMC_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key] + - ALL_GYM_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key] + - ALL_METAWORLD_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key] + key: value + ALL_DMC_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key] + ALL_GYM_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key] + ALL_METAWORLD_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key] for key, value in ALL_FANCY_MOVEMENT_PRIMITIVE_ENVIRONMENTS.items()}