Warning no longer applicable
This commit is contained in:
parent
bc9d4cf19d
commit
1c002a235b
@ -200,10 +200,6 @@ def get_env_duration(env: gym.Env):
|
|||||||
try:
|
try:
|
||||||
duration = env.spec.max_episode_steps * env.dt
|
duration = env.spec.max_episode_steps * env.dt
|
||||||
except (AttributeError, TypeError) as e:
|
except (AttributeError, TypeError) as e:
|
||||||
# TODO Remove if this information is in the compatibility class
|
|
||||||
logging.error(f'Attributes env.spec.max_episode_steps and env.dt are not available. '
|
|
||||||
f'Assuming you are using dm_control. Please make sure you have ran '
|
|
||||||
f'"pip install shimmy[dm_control]" for that.')
|
|
||||||
if env.env_type is EnvType.COMPOSER:
|
if env.env_type is EnvType.COMPOSER:
|
||||||
max_episode_steps = ceil(env.unwrapped._time_limit / env.dt)
|
max_episode_steps = ceil(env.unwrapped._time_limit / env.dt)
|
||||||
elif env.env_type is EnvType.RL_CONTROL:
|
elif env.env_type is EnvType.RL_CONTROL:
|
||||||
|
Loading…
Reference in New Issue
Block a user