From bede8723b560234f2499e1f5c4fdc230c33d2c26 Mon Sep 17 00:00:00 2001 From: RollerKnobster Date: Mon, 24 Jun 2024 15:30:02 +0300 Subject: [PATCH] Fix: remove redundant `BaseEstimator` inheritance in `KerasBaseEstimator` --- gordo/machine/model/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gordo/machine/model/models.py b/gordo/machine/model/models.py index f3d7619b6..caa985b8e 100644 --- a/gordo/machine/model/models.py +++ b/gordo/machine/model/models.py @@ -19,7 +19,7 @@ import pandas as pd import xarray as xr -from sklearn.base import TransformerMixin, BaseEstimator +from sklearn.base import TransformerMixin from sklearn.metrics import explained_variance_score from sklearn.exceptions import NotFittedError @@ -34,7 +34,7 @@ logger = logging.getLogger(__name__) -class KerasBaseEstimator(BaseWrapper, GordoBase, BaseEstimator): +class KerasBaseEstimator(BaseWrapper, GordoBase): supported_fit_args = [ "batch_size", "epochs",