2623 - finish testing action mask

tests/e2e_integration_tests/environments/test_rllib_multi_agent_environment.py

@@ -16,8 +16,6 @@ def test_rllib_multi_agent_compatibility():
     with open(MULTI_AGENT_PATH, "r") as f:
         cfg = yaml.safe_load(f)
 
-    ray.init()
-
     config = (
         PPOConfig()
         .environment(env=PrimaiteRayMARLEnv, env_config=cfg)
@@ -39,4 +37,3 @@ def test_rllib_multi_agent_compatibility():
         ),
         param_space=config,
     ).fit()
-    ray.shutdown()

tests/e2e_integration_tests/environments/test_rllib_single_agent_environment.py

@@ -20,9 +20,6 @@ def test_rllib_single_agent_compatibility():
 
     game = PrimaiteGame.from_config(cfg)
 
-    ray.shutdown()
-    ray.init()
-
     env_config = {"game": game}
     config = {
         "env": PrimaiteRayEnv,
@@ -41,4 +38,3 @@ def test_rllib_single_agent_compatibility():
     assert save_file.exists()
 
     save_file.unlink()  # clean up
-    ray.shutdown()