@@ -1,3 +1,4 @@
|
|||||||
|
import pytest
|
||||||
import ray
|
import ray
|
||||||
import yaml
|
import yaml
|
||||||
from ray import air, tune
|
from ray import air, tune
|
||||||
@@ -8,6 +9,7 @@ from primaite.game.game import PrimaiteGame
|
|||||||
from primaite.session.environment import PrimaiteRayMARLEnv
|
from primaite.session.environment import PrimaiteRayMARLEnv
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.skip(reason="Slow, reenable later")
|
||||||
def test_rllib_multi_agent_compatibility():
|
def test_rllib_multi_agent_compatibility():
|
||||||
"""Test that the PrimaiteRayEnv class can be used with a multi agent RLLIB system."""
|
"""Test that the PrimaiteRayEnv class can be used with a multi agent RLLIB system."""
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
import tempfile
|
import tempfile
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
|
import pytest
|
||||||
import ray
|
import ray
|
||||||
import yaml
|
import yaml
|
||||||
from ray.rllib.algorithms import ppo
|
from ray.rllib.algorithms import ppo
|
||||||
@@ -10,6 +11,7 @@ from primaite.game.game import PrimaiteGame
|
|||||||
from primaite.session.environment import PrimaiteRayEnv
|
from primaite.session.environment import PrimaiteRayEnv
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.skip(reason="Slow, reenable later")
|
||||||
def test_rllib_single_agent_compatibility():
|
def test_rllib_single_agent_compatibility():
|
||||||
"""Test that the PrimaiteRayEnv class can be used with a single agent RLLIB system."""
|
"""Test that the PrimaiteRayEnv class can be used with a single agent RLLIB system."""
|
||||||
with open(example_config_path(), "r") as f:
|
with open(example_config_path(), "r") as f:
|
||||||
|
|||||||
@@ -65,6 +65,7 @@ class TestPrimaiteSession:
|
|||||||
session.start_session()
|
session.start_session()
|
||||||
# TODO: include checks that the model was loaded and that the eval-only session ran
|
# TODO: include checks that the model was loaded and that the eval-only session ran
|
||||||
|
|
||||||
|
@pytest.mark.skip(reason="Slow, reenable later")
|
||||||
@pytest.mark.parametrize("temp_primaite_session", [[MULTI_AGENT_PATH]], indirect=True)
|
@pytest.mark.parametrize("temp_primaite_session", [[MULTI_AGENT_PATH]], indirect=True)
|
||||||
def test_multi_agent_session(self, temp_primaite_session):
|
def test_multi_agent_session(self, temp_primaite_session):
|
||||||
"""Check that we can run a training session with a multi agent system."""
|
"""Check that we can run a training session with a multi agent system."""
|
||||||
|
|||||||
Reference in New Issue
Block a user