Add ready rooms

StanfordVL · Jan 9, 2025 · 3970b43 · 3970b43
1 parent dc330a7
commit 3970b43
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/bddl/knowledge_base/models.py b/bddl/knowledge_base/models.py
@@ -117,7 +117,15 @@ class Scene(Model):
 
     @cached_property
     def room_count(self):
-        return len(self.rooms)
+        return len(self.future_rooms)
+
+    @cached_property
+    def ready_rooms(self):
+        return [room for room in self.rooms if room.ready]
+
+    @cached_property
+    def future_rooms(self):
+        return [room for room in self.rooms if not room.ready]
 
     @cached_property
     def object_count(self):