Merge pull request LineaLabs#558 from LineaLabs/remove-db-field-in-api

removed db access in catalog
daavoo · Mar 30, 2022 · b7e674f · b7e674f
2 parents 08bfc2e + 5f415b3
commit b7e674f
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/lineapy/graph_reader/apis.py b/lineapy/graph_reader/apis.py
@@ -160,11 +160,8 @@ class LineaCatalog:
     - The export is pretty limited right now and we should expand later.
     """
 
-    db: RelationalLineaDB
-
     def __init__(self, db):
-        self.db = db
-        db_artifacts: List[ArtifactORM] = self.db.get_all_artifacts()
+        db_artifacts: List[ArtifactORM] = db.get_all_artifacts()
         self.artifacts: List[LineaArtifact] = []
         for db_artifact in db_artifacts:
             l_artifact = LineaArtifact(