fix test problems and slowness

2024-07-12 11:23:41 +01:00
parent e759ae5990
commit 199cd0d9df
6 changed files with 13 additions and 28 deletions
--- a/tests/e2e_integration_tests/environments/test_sb3_environment.py
+++ b/tests/e2e_integration_tests/environments/test_sb3_environment.py
@@ -20,7 +20,7 @@ def test_sb3_compatibility():
    gym = PrimaiteGymEnv(env_config=cfg)
    model = PPO("MlpPolicy", gym)

-    model.learn(total_timesteps=1000)
+    model.learn(total_timesteps=256)

    save_path = Path(tempfile.gettempdir()) / "model.zip"
    model.save(save_path)