From f574a479f1d5144b1c73649fbe4ecf3f99aba48d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cem=20G=C3=B6kmen?= <cem@cemgokmen.com>
Date: Thu, 9 Jan 2025 00:39:53 -0800
Subject: [PATCH] Update __init__.py

---
 bddl/knowledge_base/__init__.py | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/bddl/knowledge_base/__init__.py b/bddl/knowledge_base/__init__.py
index 12541dd3..7ef56f72 100644
--- a/bddl/knowledge_base/__init__.py
+++ b/bddl/knowledge_base/__init__.py
@@ -10,7 +10,8 @@
   RoomRequirement, \
   RoomSynsetRequirement, \
   Room, \
-  RoomObject
+  RoomObject, \
+  AttachmentPair
 
 from bddl.knowledge_base.utils import SynsetState
 
@@ -18,3 +19,21 @@
 
 # Load the knowledge base
 KnowledgeBaseProcessor(verbose=False).run()
+
+__all__ = [
+  Property,
+  MetaLink,
+  Predicate,
+  Scene,
+  Category,
+  Object,
+  Synset,
+  TransitionRule,
+  Task,
+  RoomRequirement,
+  RoomSynsetRequirement,
+  Room,
+  RoomObject,
+  AttachmentPair,
+  SynsetState,
+]