fix

huggingface · Sep 25, 2024 · cf92fb1 · cf92fb1
1 parent eaa903e
commit cf92fb1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gym_aloha/env.py b/gym_aloha/env.py
@@ -109,7 +109,7 @@ def _make_env_task(self, task_name):
         # time limit is controlled by StepCounter in env factory
         time_limit = float("inf")
 
-        if task_name  == "transfer_cube":
+        if task_name == "transfer_cube":
             xml_path = ASSETS_DIR / "bimanual_viperx_transfer_cube.xml"
             physics = mujoco.Physics.from_xml_path(str(xml_path))
             task = TransferCubeTask()