Skip to content

Commit ec79f54

Browse files
committed
FIX: fixed failing tests, MockBasisWithWarning inherits from BaseEstimator
1 parent 3b1ad43 commit ec79f54

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/reconstruction/test_sspor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import pytest
2121
from numpy import isnan, mean, nan, sqrt, zeros
2222
from pytest_lazyfixture import lazy_fixture
23+
from sklearn.base import BaseEstimator
2324
from sklearn.exceptions import NotFittedError
2425
from sklearn.utils.validation import check_is_fitted
2526

@@ -281,7 +282,7 @@ def test_sensor_selector_properties(data_random):
281282
assert all(model.get_selected_sensors() == model.selected_sensors)
282283

283284

284-
class MockBasisWithWarning:
285+
class MockBasisWithWarning(BaseEstimator):
285286
"""Mock basis class that raises warnings when fit is called."""
286287

287288
def __init__(self, n_basis_modes=3):

0 commit comments

Comments
 (0)