Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Dec 12, 2024
1 parent ee495ad commit f6b8a14
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion omnigibson/object_states/robot_related_states.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}

0 comments on commit f6b8a14

Please sign in to comment.