first updats to reacher and env creation
This commit is contained in:
@@ -6,7 +6,7 @@ import pytest
|
||||
|
||||
from test.utils import run_env, run_env_determinism
|
||||
|
||||
CUSTOM_IDS = [spec.id for spec in gym.envs.registry.all() if
|
||||
CUSTOM_IDS = [id for id, spec in gym.envs.registry.items() if
|
||||
"fancy_gym" in spec.entry_point and 'make_bb_env_helper' not in spec.entry_point]
|
||||
CUSTOM_MP_IDS = itertools.chain(*fancy_gym.ALL_FANCY_MOVEMENT_PRIMITIVE_ENVIRONMENTS.values())
|
||||
SEED = 1
|
||||
|
||||
@@ -24,6 +24,7 @@ def run_env(env_id, iterations=None, seed=0, render=False):
|
||||
actions = []
|
||||
dones = []
|
||||
obs = env.reset()
|
||||
print(obs.dtype)
|
||||
verify_observations(obs, env.observation_space, "reset()")
|
||||
|
||||
iterations = iterations or (env.spec.max_episode_steps or 1)
|
||||
|
||||
Reference in New Issue
Block a user