minor fix

BorgwardtLab · Apr 25, 2024 · fd777d6 · fd777d6
1 parent d2c5c63
commit fd777d6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/proteinshake/task.py b/proteinshake/task.py
@@ -62,14 +62,14 @@ def split(self, Xy):
 
     def loader(
         self,
-        split=None,
+        split=0,
         batch_size=None,
         shuffle: bool = False,
         random_seed: Union[int, None] = None,
         **kwargs,
     ):
         rng = np.random.default_rng(random_seed)
-        path = self.root / split / self.transform.hash / "shards"
+        path = self.root / split / hash(self.transform) / "shards"
         shard_index = load(path / "index.npy")
         if (
             not batch_size is None

diff --git a/proteinshake/transform.py b/proteinshake/transform.py
@@ -86,8 +86,7 @@ def __init__(self, *transforms):
                     error("You cannot use more than one framework.")
                 setattr(self, "create_loader", transform.create_loader)
 
-    @property
-    def hash(self):
+    def __hash___(self):
         return self.__class__.__name__
 
     def fit(self, dataset):