Remove redundant inheritance

ShoggothAI · Apr 25, 2024 · e6ff0a1 · e6ff0a1
1 parent dc0c4c9
commit e6ff0a1
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/motleycrew/storage/kuzu_graph_store.py b/motleycrew/storage/kuzu_graph_store.py
@@ -5,12 +5,10 @@
 
 from typing import Any, Dict, List, Optional
 
-from llama_index.core.graph_stores.types import GraphStore
-
 import kuzu
 
 
-class MotleyQuestionGraphStore(GraphStore):
+class MotleyQuestionGraphStore:
     IS_SUBQUESTION_PREDICATE = "IS_SUBQUESTION"
 
     def __init__(
@@ -175,7 +173,6 @@ def from_dict(cls, config_dict: Dict[str, Any]) -> "MotleyQuestionGraphStore":
 
 if __name__ == "__main__":
     from pathlib import Path
-    import kuzu
 
     here = Path(__file__).parent
     db_path = here / "test1"