From 930e2e54ff7243ba5d7ab6f65c9409e5e21a2b62 Mon Sep 17 00:00:00 2001
From: George Yiasemis <georgeyiasemis@hotmail.com>
Date: Sun, 20 Oct 2024 22:45:53 +0200
Subject: [PATCH] Minor fix

---
 direct/nn/transformers/mri_transformers.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/direct/nn/transformers/mri_transformers.py b/direct/nn/transformers/mri_transformers.py
index dd2ab81f..65c2f06e 100644
--- 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`.