Remove defaults.

molecularmachines · May 19, 2024 · 3ea2e27 · 3ea2e27
1 parent abcf352
commit 3ea2e27
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/moleculib/protein/dataset.py b/moleculib/protein/dataset.py
@@ -486,16 +486,19 @@ class TimewarpDataset(torch.utils.data.Dataset):
 
     def __init__(
         self,
-        dataset: str = "2AA-1-big",
-        split: str = "train",
-        tau: int = 1,
+        dataset: str,
+        split: str,
+        tau: int,
     ):
         base = "/mas/projects/molecularmachines/db/timewarp2/"
         self.base_path = os.path.join(base, dataset, split)
         self.counter = 0
         self.tau = tau
 
         self.files = self._list_files()
+        if len(self.files) == 0:
+            raise ValueError(f"No files found in {self.base_path}")
+
         print(f"Found {len(self.files)} files in {self.base_path}")
 
         self._load_coords(self.files[0])