fix flaky tests

StanfordVL · Dec 2, 2024 · e253686 · e253686
1 parent a02bb6e
commit e253686
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tests/test_data_collection.py b/tests/test_data_collection.py
@@ -29,7 +29,7 @@ def test_data_collect_and_playback():
         "task": {
             "type": "BehaviorTask",
             # BehaviorTask-specific
-            "activity_name": "assembling_gift_baskets",
+            "activity_name": "laying_wood_floors",
             "online_object_sampling": True,
         },
     }

diff --git a/tests/test_robot_states_no_flatcache.py b/tests/test_robot_states_no_flatcache.py
@@ -39,7 +39,8 @@ def test_object_in_FOV_of_robot():
     vision_sensor = sensors[0]
     vision_sensor.set_position_orientation(position=[100, 150, 100])
     og.sim.step()
-    og.sim.step()
+    for _ in range(5):
+        og.sim.render()
     # Since the sensor is moved away from the robot, the robot should not see itself
     assert robot.states[ObjectsInFOVOfRobot].get_value() == []
     og.clear()