Add forgotten argument

NKI-AI · Jun 6, 2024 · fbad9e1 · fbad9e1
1 parent c6b4c0a
commit fbad9e1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/direct/data/datasets_config.py b/direct/data/datasets_config.py
@@ -92,6 +92,9 @@ class TransformsConfig(BaseConfig):
         Default is True.
     image_recon_type : ReconstructionType
         Image reconstruction type. Default is ReconstructionType.RSS.
+    compress_coils : int, optional
+        Number of coils to compress input k-space. It is not recommended to be used in combination with `pad_coils`.
+        Default is None.
     pad_coils : int, optional
         Pad coils. Default is None.
     use_seed : bool
@@ -133,6 +136,7 @@ class TransformsConfig(BaseConfig):
     delete_acs_mask: bool = True
     delete_kspace: bool = True
     image_recon_type: ReconstructionType = ReconstructionType.RSS
+    compress_coils: Optional[int] = None
     pad_coils: Optional[int] = None
     use_seed: bool = True
     transforms_type: TransformsType = TransformsType.SUPERVISED