From ea37c8fecc1f534aafca1f8d65acd89579db1d59 Mon Sep 17 00:00:00 2001
From: JasperVS <jmvanselm@gmail.com>
Date: Fri, 27 Dec 2024 23:48:57 +0100
Subject: [PATCH] Remove unnecessary noqa

---
 epochlib/model.py                                     | 4 ++--
 epochlib/training/augmentation/image_augmentations.py | 2 +-
 epochlib/transformation/transformation.py             | 2 +-
 epochlib/transformation/transformation_block.py       | 4 ++--
 4 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/epochlib/model.py b/epochlib/model.py
index f7f7a5c..f2c67b2 100644
--- a/epochlib/model.py
+++ b/epochlib/model.py
@@ -23,7 +23,7 @@ def __post_init__(self) -> None:
         """
         return super().__post_init__()
 
-    def train(self, x: Any, y: Any, **train_args: Any) -> tuple[Any, Any]:  # noqa: ANN401
+    def train(self, x: Any, y: Any, **train_args: Any) -> tuple[Any, Any]:
         """Train the system.
 
         :param x: The input to the system.
@@ -32,7 +32,7 @@ def train(self, x: Any, y: Any, **train_args: Any) -> tuple[Any, Any]:  # noqa:
         """
         return super().train(x, y, **train_args)
 
-    def predict(self, x: Any, **pred_args: Any) -> Any:  # noqa: ANN401
+    def predict(self, x: Any, **pred_args: Any) -> Any:
         """Predict the output of the system.
 
         :param x: The input to the system.
diff --git a/epochlib/training/augmentation/image_augmentations.py b/epochlib/training/augmentation/image_augmentations.py
index bf042f7..86c27a1 100644
--- a/epochlib/training/augmentation/image_augmentations.py
+++ b/epochlib/training/augmentation/image_augmentations.py
@@ -6,7 +6,7 @@
 import torch
 
 
-def get_kornia_mix() -> Any:  # noqa: ANN401
+def get_kornia_mix() -> Any:
     """Return kornia mix."""
     try:
         import kornia
diff --git a/epochlib/transformation/transformation.py b/epochlib/transformation/transformation.py
index 41b017d..213814e 100644
--- a/epochlib/transformation/transformation.py
+++ b/epochlib/transformation/transformation.py
@@ -59,7 +59,7 @@ def log_to_terminal(self, message: str) -> None:
 
     title: str = "Transformation Pipeline"  # The title of the pipeline since transformation pipeline can be used for multiple purposes. (Feature, Label, etc.)
 
-    def transform(self, data: Any, cache_args: CacheArgs | None = None, **transform_args: Any) -> Any:  # noqa: ANN401
+    def transform(self, data: Any, cache_args: CacheArgs | None = None, **transform_args: Any) -> Any:
         """Transform the input data.
 
         :param data: The input data.
diff --git a/epochlib/transformation/transformation_block.py b/epochlib/transformation/transformation_block.py
index e311235..fbe82cf 100644
--- a/epochlib/transformation/transformation_block.py
+++ b/epochlib/transformation/transformation_block.py
@@ -54,7 +54,7 @@ def custom_transform(self, data: Any) -> Any:
         data = custom_transformation_block.transform(data, cache=cache_args)
     """
 
-    def transform(self, data: Any, cache_args: CacheArgs | None = None, **transform_args: Any) -> Any:  # noqa: ANN401
+    def transform(self, data: Any, cache_args: CacheArgs | None = None, **transform_args: Any) -> Any:
         """Transform the input data using a custom method.
 
         :param data: The input data.
@@ -77,7 +77,7 @@ def transform(self, data: Any, cache_args: CacheArgs | None = None, **transform_
         return data
 
     @abstractmethod
-    def custom_transform(self, data: Any, **transform_args: Any) -> Any:  # noqa: ANN401
+    def custom_transform(self, data: Any, **transform_args: Any) -> Any:
         """Transform the input data using a custom method.
 
         :param data: The input data.