Fix: remove redundant BaseEstimator inheritance in `KerasBaseEstima…

…tor`
equinor · Jun 24, 2024 · bede872 · bede872
1 parent 77bc40c
commit bede872
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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",