Update __init__.py

StanfordVL · Jan 9, 2025 · f574a47 · f574a47
1 parent 174c698
commit f574a47
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/bddl/knowledge_base/__init__.py b/bddl/knowledge_base/__init__.py
@@ -10,11 +10,30 @@
   RoomRequirement, \
   RoomSynsetRequirement, \
   Room, \
-  RoomObject
+  RoomObject, \
+  AttachmentPair
 
 from bddl.knowledge_base.utils import SynsetState
 
 from bddl.knowledge_base.processing import KnowledgeBaseProcessor
 
 # Load the knowledge base
 KnowledgeBaseProcessor(verbose=False).run()
+
+__all__ = [
+  Property,
+  MetaLink,
+  Predicate,
+  Scene,
+  Category,
+  Object,
+  Synset,
+  TransitionRule,
+  Task,
+  RoomRequirement,
+  RoomSynsetRequirement,
+  Room,
+  RoomObject,
+  AttachmentPair,
+  SynsetState,
+]