Update datasets.py

shenqq377 · Dec 5, 2023 · 70a828d · 70a828d
1 parent 7ac2ce6
commit 70a828d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataloaders/datasets.py b/dataloaders/datasets.py
@@ -28,7 +28,7 @@ def __init__(self, args):
 
         self.image_dirs = sorted(self.image_dirs, key=lambda x: int(x.split('_')[-1].split('.nii.gz')[0]))
 
-        # remove test fold!
+        # test fold!
         self.FOLD = get_folds(args['dataset'])
         self.image_dirs = [elem for idx, elem in enumerate(self.image_dirs) if idx in self.FOLD[args['eval_fold']]]