Add contrib

huawei-noah · Jul 30, 2022 · acdb102 · acdb102
1 parent 40f3c7f
commit acdb102
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -0,0 +1,3 @@
+[submodule "competition/contrib/smarts-project.bubble_env"]
+	path = competition/contrib/smarts-project.bubble_env
+	url = https://github.com/smarts-project/smarts-project.bubble_env
diff --git a/competition/contrib/smarts-project.bubble_env b/competition/contrib/smarts-project.bubble_env
diff --git a/smarts/env/multi_scenario_v0_env.py b/smarts/env/multi_scenario_v0_env.py
@@ -109,7 +109,7 @@ def multi_scenario_v0_env(
 
     agent_specs = {
         f"Agent_{i}": AgentSpec(
-            interface=_resolve_agent_interface(img_meters, img_pixels, action_space)
+            interface=resolve_agent_interface(img_meters, img_pixels, action_space)
         )
         for i in range(env_specs["num_agent"])
     }
@@ -238,7 +238,7 @@ def _get_env_specs(scenario: str):
         raise Exception(f"Unknown scenario {scenario}.")
 
 
-def _resolve_agent_interface(
+def resolve_agent_interface(
     img_meters: int = 64, img_pixels: int = 256, action_space="TargetPose", **kwargs
 ):
     done_criteria = DoneCriteria(