From 181fdb0b2f05a0d89a775044f22f889fa3697855 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nguy=E1=BB=85n=20Huy=20Ho=C3=A0ng?= <115699781+hoanghelloworld@users.noreply.github.com> Date: Tue, 13 May 2025 23:56:03 +0700 Subject: [PATCH] Fix typo in file filtering condition in DEFAULTDataset --- dataset/default.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dataset/default.py b/dataset/default.py index 28bb0bd..7258364 100644 --- a/dataset/default.py +++ b/dataset/default.py @@ -26,7 +26,7 @@ def __init__(self, root_dir: str): def get_data_files(self): nifti_file_names = os.listdir(self.root_dir) folder_names = [os.path.join( - self.root_dir, nifti_file_name) for nifti_file_name in nifti_file_names if nifti_file_names.endsiwth('.nii')] + self.root_dir, nifti_file_name) for nifti_file_name in nifti_file_names if nifti_file_names.endswith('.nii')] return folder_names def __len__(self):