Remove unnecessary noqa

TeamEpochGithub · Dec 27, 2024 · ea37c8f · ea37c8f
1 parent c27d5ab
commit ea37c8f
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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.