limit arg for Database.load

saezlab · Nov 20, 2024 · 9666016 · 9666016
1 parent 630b15d
commit 9666016
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/omnipath_metabo/schema/_main.py b/omnipath_metabo/schema/_main.py
@@ -43,10 +43,10 @@ def create(self) -> None:
         create(self.con)
 
 
-    def load(self, resource) -> None:
+    def load(self, resource: str, limit: int = None) -> None:
 
         loader = Loader(resource, self.con)
-        loader.load()
+        loader.load(limit = limit)
 
     def test_load(self, resource) -> None: