Speed up rllib test (#1767)

huawei-noah · Dec 6, 2022 · 9904e94 · 9904e94
1 parent f90705a
commit 9904e94
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/smarts/env/tests/test_rllib_hiway_env.py b/smarts/env/tests/test_rllib_hiway_env.py
@@ -36,6 +36,7 @@
 from smarts.core.utils.file import make_dir_in_smarts_log_dir
 from smarts.env.rllib_hiway_env import RLlibHiWayEnv
 from smarts.zoo.agent_spec import AgentSpec
+from smarts import sstudio
 
 AGENT_ID = "Agent-007"
 INFO_EXTRA_KEY = "__test_extra__"
@@ -124,7 +125,9 @@ def on_episode_end(info):
     }
 
     # XXX: We should be able to simply provide "scenarios/sumo/loop"?
-    scenario_path = Path(__file__).parent / "../../../scenarios/sumo/loop"
+    import smarts
+    scenario_path = Path(smarts.__file__).parent.absolute()/"scenarios/intersection/1_to_1lane_left_turn_c"
+
     tune_confg = {
         "env": RLlibHiWayEnv,
         "env_config": {