Modify tests based on refactoring

2024-02-26 12:15:53 +00:00
parent ccb10f1160
commit a5043a8fbe
16 changed files with 28 additions and 174 deletions
--- a/tests/e2e_integration_tests/environments/test_sb3_environment.py
+++ b/tests/e2e_integration_tests/environments/test_sb3_environment.py
@@ -17,8 +17,7 @@ def test_sb3_compatibility():
    with open(example_config_path(), "r") as f:
        cfg = yaml.safe_load(f)

-    game = PrimaiteGame.from_config(cfg)
-    gym = PrimaiteGymEnv(game=game)
+    gym = PrimaiteGymEnv(game_config=cfg)
    model = PPO("MlpPolicy", gym)

    model.learn(total_timesteps=1000)