diff --git a/omnigibson/object_states/robot_related_states.py b/omnigibson/object_states/robot_related_states.py index 1e38265ae..6567951d8 100644 --- a/omnigibson/object_states/robot_related_states.py +++ b/omnigibson/object_states/robot_related_states.py @@ -64,6 +64,12 @@ def _get_value(self): for sensor in self.robot.sensors.values(): if isinstance(sensor, VisionSensor): _, info = sensor.get_obs() - objs.update(set(self.obj.scene.object_registry("name", name) for name in info["seg_instance"].values() if name not in names_to_exclude)) + objs.update( + set( + self.obj.scene.object_registry("name", name) + for name in info["seg_instance"].values() + if name not in names_to_exclude + ) + ) # Return all objects, minus any that were mapped to None because they were not found in our object registry return objs - {None}