Fix some bugs in knowledgebase.

StanfordVL · Oct 3, 2023 · 23bba90 · 23bba90
1 parent eb321c1
commit 23bba90
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/bddl/knowledge_base/models.py b/bddl/knowledge_base/models.py
@@ -102,7 +102,7 @@ def room_count(self):
     @cached_property
     def object_count(self):
         return sum(
-            roomobject.object_count
+            roomobject.count
             for room in self.rooms
             for roomobject in room.roomobjects
             if not room.ready
@@ -115,7 +115,7 @@ def any_ready(self):
     @cached_property
     def fully_ready(self):
         ready_count = sum(
-            roomobject.object_count
+            roomobject.count
             for room in self.rooms
             for roomobject in room.roomobjects
             if room.ready
@@ -482,7 +482,7 @@ def synset_state(self) -> str:
             return STATE_UNMATCHED
         elif any(synset.state == STATE_UNMATCHED for synset in self.synsets):
             return STATE_UNMATCHED
-        elif any(synset.state == STATE_MATCHED for synset in self.synsets):
+        elif any(synset.state == STATE_PLANNED for synset in self.synsets):
             return STATE_PLANNED
         else:
             return STATE_MATCHED