dodox/fancy_gym
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Improve handling of missing Metaworld

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-10-12 15:34:36 +02:00
parent a5b10e11a3
commit 0c93280796
2 changed files with 56 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
fancy_gym/meta/__init__.py
View File

@ -7,6 +7,13 @@ from ..envs.registry import register
from . import goal_object_change_mp_wrapper, goal_change_mp_wrapper, goal_endeffector_change_mp_wrapper, \ from . import goal_object_change_mp_wrapper, goal_change_mp_wrapper, goal_endeffector_change_mp_wrapper, \
object_change_mp_wrapper object_change_mp_wrapper
try:
import metaworld
except ModuleNotFoundError:
print('[FANCY GYM] Metaworld not avaible.')
else:
# Will only get executed, if import succeeds
from . import metaworld_adapter from . import metaworld_adapter
metaworld_adapter.register_all_ML1() metaworld_adapter.register_all_ML1()
@ -60,3 +67,4 @@ for _task in _goal_and_endeffector_change_envs:
mp_wrapper=goal_endeffector_change_mp_wrapper.MPWrapper, mp_wrapper=goal_endeffector_change_mp_wrapper.MPWrapper,
add_mp_types=['ProMP', 'ProDMP'], add_mp_types=['ProMP', 'ProDMP'],
) )
`

4
fancy_gym/meta/metaworld_adapter.py
View File

@ -11,11 +11,7 @@ import numpy as np
from fancy_gym.utils.env_compatibility import EnvCompatibility from fancy_gym.utils.env_compatibility import EnvCompatibility
try:
import metaworld import metaworld
except Exception:
print('[FANCY GYM] Metaworld not avaible')
class FixMetaworldHasIncorrectObsSpaceWrapper(gym.Wrapper, gym.utils.RecordConstructorArgs): class FixMetaworldHasIncorrectObsSpaceWrapper(gym.Wrapper, gym.utils.RecordConstructorArgs):
def __init__(self, env: gym.Env): def __init__(self, env: gym.Env):