diff --git a/src/pyopmnearwell/ml/scaler_layers.py b/src/pyopmnearwell/ml/scaler_layers.py
index d3b6a84..a9c7924 100644
--- a/src/pyopmnearwell/ml/scaler_layers.py
+++ b/src/pyopmnearwell/ml/scaler_layers.py
@@ -4,8 +4,6 @@
 linting completely. It is possible that the module is not functional at the moment.
 """
 
-# pylint: skip-file
-
 from __future__ import annotations
 
 from typing import Optional, Sequence, Union
@@ -15,6 +13,7 @@
 import tensorflow as tf
 from numpy.typing import ArrayLike
 
+# pylint: skip-file
 
 class ScalerLayer(keras.layers.Layer):
     """MixIn to provide functionality for the Scaler Layer."""
diff --git a/tests/test_scaler_layers.py b/tests/test_scaler_layers.py
index 9a90916..592227d 100644
--- a/tests/test_scaler_layers.py
+++ b/tests/test_scaler_layers.py
@@ -18,6 +18,8 @@
 
 from pyopmnearwell.ml.scaler_layers import MinMaxScalerLayer, MinMaxUnScalerLayer
 
+# pylint: skip-file
+
 rng: np.random.Generator = np.random.default_rng()
 
 layers: list[str] = ["scaler", "unscaler"]