#2588 optional RL deps

2024-05-29 14:54:45 +01:00
parent 85f03570f7
commit 0c19546322
14 changed files with 201 additions and 182 deletions
--- a/tests/integration_tests/configuration_file_parsing/test_episode_scheduler.py
+++ b/tests/integration_tests/configuration_file_parsing/test_episode_scheduler.py
@@ -1,7 +1,8 @@
 import pytest
 import yaml

-from primaite.session.environment import PrimaiteGymEnv, PrimaiteRayEnv, PrimaiteRayMARLEnv
+from primaite.session.environment import PrimaiteGymEnv
+from primaite.session.ray_envs import PrimaiteRayEnv, PrimaiteRayMARLEnv
 from tests.conftest import TEST_ASSETS_ROOT

 folder_path = TEST_ASSETS_ROOT / "configs" / "scenario_with_placeholders"