hotfix missing split fallback

BorgwardtLab · Jul 3, 2024 · a5aa263 · a5aa263
1 parent c6d578a
commit a5aa263
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/proteinshake/tasks/task.py b/proteinshake/tasks/task.py
@@ -84,6 +84,7 @@ def compute_index(self):
             self.val_index = np.array([i for i,p in enumerate(self.proteins) if p['protein'][split_name] == 'val'])
             self.test_index = np.array([i for i,p in enumerate(self.proteins) if p['protein'][split_name] == 'test'])
         else:
+            print('Could not find pre-computed split. Falling back to compute_custom_split, which by default is a random split.')
             self.train_index, self.val_index, self.test_index = self.compute_custom_split(self.split)
 
         self.update_index()