Fix unit test

Signed-off-by: Beat Buesser <[email protected]>

Author: beat-buesser

art/estimators/classification/keras.py

@@ -559,7 +559,7 @@ def predict(  # pylint: disable=W0221
 
         return predictions
 
-    def fit(
+    def fit(  # pylint: disable=W0221
         self, x: np.ndarray, y: np.ndarray, batch_size: int = 128, nb_epochs: int = 20, verbose: bool = False, **kwargs
     ) -> None:
         """
@@ -589,7 +589,9 @@ def fit(
             x=x_preprocessed, y=y_preprocessed, batch_size=batch_size, epochs=nb_epochs, verbose=int(verbose), **kwargs
         )
 
-    def fit_generator(self, generator: "DataGenerator", nb_epochs: int = 20, verbose: bool = False, **kwargs) -> None:
+    def fit_generator(  # pylint: disable=W0221
+        self, generator: "DataGenerator", nb_epochs: int = 20, verbose: bool = False, **kwargs
+    ) -> None:
         """
         Fit the classifier using the generator that yields batches as specified.