first updats to reacher and env creation

This commit is contained in:
Fabian
2022-10-20 10:10:44 +02:00
parent 14ac5f81c7
commit 0c7ac838bf
4 changed files with 34 additions and 8 deletions
+1 -1
View File
@@ -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
+1
View File
@@ -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)