linter

Safe-DS · Aug 19, 2024 · 7b66cdd · 7b66cdd
1 parent 7458e75
commit 7b66cdd
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/safeds/ml/nn/_model.py b/src/safeds/ml/nn/_model.py
@@ -962,8 +962,8 @@ def fit_by_exhaustive_search(
         elif isinstance(train_data, TimeSeriesDataset):
             return self._get_best_rnn_model(
                 list_of_fitted_models,
-                train_set,
-                test_set,
+                train_set,  # type: ignore[arg-type]
+                test_set,   # type: ignore[arg-type]
                 optimization_metric,  # type: ignore[arg-type]
                 positive_class,
             )  # type: ignore[arg-type]
@@ -1185,8 +1185,8 @@ def _get_best_cnn_model_column(
         best_metric_value = None
         for fitted_model in list_of_fitted_models:
             prediction = self._inverse_one_hot_encode_by_index_of_column(
-                fitted_model.predict(input_data).get_output(),
-            )  # type: ignore[attr-defined, arg-type]
+                fitted_model.predict(input_data).get_output(),  # type: ignore[attr-defined, arg-type]
+            )
             if best_model is None:
                 best_model = fitted_model
                 match optimization_metric:
@@ -1259,8 +1259,8 @@ def _get_best_cnn_model_table(
         best_metric_value = None
         for fitted_model in list_of_fitted_models:
             prediction = self._inverse_one_hot_encode_by_index_of_column(
-                fitted_model.predict(input_data).get_output(),
-            )  # type: ignore[attr-defined, arg-type]
+                fitted_model.predict(input_data).get_output(),  # type: ignore[attr-defined, arg-type]
+            )
             if best_model is None:
                 best_model = fitted_model
                 match optimization_metric: