#2476 Add test for episode scheduler

2024-04-25 15:09:46 +01:00
parent 42ce264e73
commit 66f31e8ed1
21 changed files with 456 additions and 23 deletions
--- a/tests/e2e_integration_tests/environments/test_sb3_environment.py
+++ b/tests/e2e_integration_tests/environments/test_sb3_environment.py
@@ -16,7 +16,7 @@ def test_sb3_compatibility():
    with open(data_manipulation_config_path(), "r") as f:
        cfg = yaml.safe_load(f)

-    gym = PrimaiteGymEnv(game_config=cfg)
+    gym = PrimaiteGymEnv(env_config=cfg)
    model = PPO("MlpPolicy", gym)

    model.learn(total_timesteps=1000)