Improve handling of missing Metaworld
This commit is contained in:
parent
a5b10e11a3
commit
0c93280796
@ -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, \
|
||||
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
|
||||
|
||||
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,
|
||||
add_mp_types=['ProMP', 'ProDMP'],
|
||||
)
|
||||
`
|
@ -11,11 +11,7 @@ import numpy as np
|
||||
|
||||
from fancy_gym.utils.env_compatibility import EnvCompatibility
|
||||
|
||||
try:
|
||||
import metaworld
|
||||
except Exception:
|
||||
print('[FANCY GYM] Metaworld not avaible')
|
||||
|
||||
|
||||
class FixMetaworldHasIncorrectObsSpaceWrapper(gym.Wrapper, gym.utils.RecordConstructorArgs):
|
||||
def __init__(self, env: gym.Env):
|
||||
|
Loading…
Reference in New Issue
Block a user