Enums to uppercase

NKI-AI · Mar 24, 2024 · 97d8909 · 97d8909
1 parent b56a724
commit 97d8909
Show file tree

Hide file tree

Showing 16 changed files with 272 additions and 221 deletions.
diff --git a/direct/algorithms/mri_algorithms.py b/direct/algorithms/mri_algorithms.py
@@ -37,7 +37,7 @@ def __init__(
         kernel_size: int = 6,
         crop: float = 0.95,
         max_iter: int = 100,
-        kspace_key: KspaceKey = KspaceKey.masked_kspace,
+        kspace_key: KspaceKey = KspaceKey.MASKED_KSPACE,
     ):
         """Inits :class:`EstimateSensitivityMap`.
 
@@ -54,7 +54,7 @@ def __init__(
         max_iter: int, optional
             Power method iterations. Default: 30.
         kspace_key: KspaceKey
-            K-space key. Default KspaceKey.masked_kspace.
+            K-space key. Default KspaceKey.MASKED_KSPACE.
         """
         self.backward_operator = backward_operator
         self.threshold = threshold

diff --git a/direct/data/datasets_config.py b/direct/data/datasets_config.py
@@ -32,12 +32,10 @@ class SensitivityMapEstimationTransformConfig(BaseConfig):
 
 @dataclass
 class RandomAugmentationTransformsConfig(BaseConfig):
-    random_rotation: bool = False
     random_rotation_degrees: Tuple[int, ...] = (-90, 90)
-    random_rotation_probability: Optional[float] = 0.5
-    random_flip: bool = False
+    random_rotation_probability: float = 0.0
     random_flip_type: Optional[str] = "random"
-    random_flip_probability: Optional[float] = 0.5
+    random_flip_probability: float = 0.0
 
 
 @dataclass