bug

Author: jf-garamendi

ingestion/no_mask_with_generic_sequence_multiscale.py

@@ -30,7 +30,7 @@ def __getitem__(self, idx):
         # take a random mask sequence
         n_mask_seq = random.randint(0, len(self.mask_folders)-1)
 
-        if idx > len(self.mask_folders):
+        if idx > len(self.mask_folders)-1:
             masks_folder = join(self.generic_mask_sequences_dir, self.mask_folders[0])
         else:
             masks_folder = join(self.generic_mask_sequences_dir, self.mask_folders[idx])