Minor fix

NKI-AI · Oct 20, 2024 · 930e2e5 · 930e2e5
1 parent d14e00f
commit 930e2e5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/direct/nn/transformers/mri_transformers.py b/direct/nn/transformers/mri_transformers.py
@@ -78,6 +78,7 @@ def __init__(
         locality_strength: float = 1.0,
         use_pos_embedding: bool = True,
         normalized: bool = True,
+        **kwargs,
     ) -> None:
         """Inits :class:`ImageDomainMRIViT2D`.
 
@@ -231,6 +232,7 @@ def __init__(
         locality_strength: float = 1.0,
         use_pos_embedding: bool = True,
         normalized: bool = True,
+        **kwargs,
     ) -> None:
         """Inits :class:`ImageDomainMRIViT3D`.
 
@@ -386,6 +388,7 @@ def __init__(
         use_pos_embedding: bool = True,
         normalized: bool = True,
         compute_per_coil: bool = True,
+        **kwargs,
     ) -> None:
         """Inits :class:`KSpaceDomainMRIViT2D`.
 
@@ -575,6 +578,7 @@ def __init__(
         use_pos_embedding: bool = True,
         normalized: bool = True,
         compute_per_coil: bool = True,
+        **kwargs,
     ) -> None:
         """Inits :class:`KSpaceDomainMRIViT3D`.