Minor fix

NKI-AI · Apr 19, 2024 · 924bc5f · 924bc5f
1 parent 2572474
commit 924bc5f
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/direct/data/datasets_config.py b/direct/data/datasets_config.py
@@ -2,8 +2,6 @@
 
 """Classes holding the typed configurations for the datasets."""
 
-from __future__ import annotations
-
 from dataclasses import dataclass
 from typing import List, Optional, Tuple
 
@@ -149,17 +147,17 @@ class H5SliceConfig(DatasetConfig):
     pass_mask: bool = False
     data_root: Optional[str] = None
     filenames_filter: Optional[List[str]] = None
-    filenames_Lists: Optional[List[str]] = None
-    filenames_Lists_root: Optional[str] = None
+    filenames_lists: Optional[List[str]] = None
+    filenames_lists_root: Optional[str] = None
 
 
 @dataclass
 class CMRxReconConfig(DatasetConfig):
     regex_filter: Optional[str] = None
     data_root: Optional[str] = None
     filenames_filter: Optional[List[str]] = None
-    filenames_Lists: Optional[List[str]] = None
-    filenames_Lists_root: Optional[str] = None
+    filenames_lists: Optional[List[str]] = None
+    filenames_lists_root: Optional[str] = None
     kspace_key: str = "kspace_full"
     compute_mask: bool = False
     extra_keys: Optional[List[str]] = None