Enums to uppercase

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

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