Codacy/pylint fixes

NKI-AI · Jan 7, 2025 · 43c5dab · 43c5dab
1 parent ccb7088
commit 43c5dab
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/direct/nn/transformers/transformers.py b/direct/nn/transformers/transformers.py
@@ -1,5 +1,7 @@
 # Copyright (c) DIRECT Contributors
 
+# pylint: disable=too-many-arguments
+
 """DIRECT Vision Transformer models for MRI reconstruction."""
 
 from __future__ import annotations
@@ -162,7 +164,6 @@ def __init__(
             Whether to apply normalization before and denormalization after the forward pass. Default: True.
         **kwargs: Other keyword arguments to pass to the parent constructor.
         """
-        # pylint: disable=too-many-arguments
         super().__init__()
         for extra_key in kwargs:
             if extra_key not in [

diff --git a/direct/nn/transformers/uformer.py b/direct/nn/transformers/uformer.py
@@ -1653,6 +1653,7 @@ def __init__(
             Whether to use cross-modulation in the attention mechanism. Default: False.
         **kwargs: Other keyword arguments to pass to the parent constructor.
         """
+        # pylint: disable=too-many-locals
         super().__init__()
         if len(encoder_num_heads) != len(encoder_depths):
             raise ValueError(
@@ -1985,6 +1986,7 @@ def __init__(
             Whether to apply normalization before and denormalization after the forward pass. Default: True.
         **kwargs: Other keyword arguments to pass to the parent constructor.
         """
+        # pylint: disable=too-many-locals
         super().__init__()
 
         self.uformer = UFormer(