Update lazyloading.py

chenyangkang · Oct 27, 2024 · 08ac586 · 08ac586
1 parent 6c313a3
commit 08ac586
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/stemflow/utils/lazyloading.py b/stemflow/utils/lazyloading.py
@@ -7,7 +7,10 @@ class LazyLoadingEnsembleDict(MutableMapping):
     def __init__(self, directory='./tmp_models'):
         """
         Initialize the LazyLoadingEnsembleDict with a directory to save and load models.
-        :param directory: The directory to save and load model files.
+        
+        Args:
+            directory:
+                The directory to save and load model files.
         """
         self.directory = directory
         self.ensemble_models = {}
@@ -197,4 +200,4 @@ def delete_ensemble(self, ensemble_id):
         if os.path.exists(ensemble_path):
             os.remove(ensemble_path)
         else:
-            raise ValueError(f'Ensemble {ensemble_id} not found on disk.')
+            raise ValueError(f'Ensemble {ensemble_id} not found on disk.')