Minor fix

NKI-AI · Feb 5, 2024 · 2829dc8 · 2829dc8
1 parent e95b890
commit 2829dc8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/direct/common/subsample_config.py b/direct/common/subsample_config.py
@@ -5,7 +5,7 @@
 
 from omegaconf import MISSING
 
-from direct.config.defaults import BaseConfig
+from direct.config import BaseConfig
 
 
 @dataclass

diff --git a/direct/data/datasets_config.py b/direct/data/datasets_config.py
@@ -11,7 +11,7 @@
 from omegaconf import MISSING
 
 from direct.common.subsample_config import MaskingConfig
-from direct.config.defaults import BaseConfig
+from direct.config import BaseConfig
 from direct.data.transforms import RescaleMode
 
 
@@ -71,6 +71,7 @@ class TransformsConfig(BaseConfig):
     image_recon_type: str = "rss"
     pad_coils: Optional[int] = None
     use_seed: bool = True
+    dynamic_mask: bool = False
     target_accelerations: Optional[tuple[float, ...]] = None