Refactor eval.py (#127)

2024-05-03 17:33:16 +01:00
parent b7b69fcc3d
commit bccee745c3
12 changed files with 457 additions and 298 deletions
--- a/tests/test_envs.py
+++ b/tests/test_envs.py
@@ -37,7 +37,7 @@ def test_factory(env_name):
        overrides=[f"env={env_name}", f"device={DEVICE}"],
    )

-    env = make_env(cfg, num_parallel_envs=1)
+    env = make_env(cfg, n_envs=1)
    obs, _ = env.reset()
    obs = preprocess_observation(obs)