Skip to content

Commit b5e6fea

Browse files
authored
Remove mention of former compat modules (#1588)
* Remove mention of former `compat` modules * Format test_estimators.py
1 parent d606d7b commit b5e6fea

File tree

2 files changed

+1
-18
lines changed

2 files changed

+1
-18
lines changed

river/conftest.py

-6
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
try:
66
import sklearn # noqa: F401
77
except ImportError:
8-
collect_ignore.append("compat/sklearn.py")
98
collect_ignore.append("compat/test_sklearn.py")
109

1110
try:
@@ -14,11 +13,6 @@
1413
collect_ignore.append("stream/iter_sql.py")
1514
collect_ignore.append("stream/test_sql.py")
1615

17-
try:
18-
import torch # noqa: F401
19-
except ImportError:
20-
collect_ignore.append("compat/pytorch.py")
21-
2216
try:
2317
import vaex # noqa: F401
2418
except ImportError:

river/test_estimators.py

+1-12
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import inspect
66

77
import pytest
8+
from sklearn import linear_model as sk_linear_model
89

910
from river import (
1011
anomaly,
@@ -24,15 +25,6 @@
2425
preprocessing,
2526
time_series,
2627
)
27-
28-
try:
29-
from river.compat.pytorch import PyTorch2RiverBase
30-
31-
PYTORCH_INSTALLED = True
32-
except ImportError:
33-
PYTORCH_INSTALLED = False
34-
from sklearn import linear_model as sk_linear_model
35-
3628
from river.compat.river_to_sklearn import River2SKLBase
3729
from river.compat.sklearn_to_river import SKL2RiverBase
3830

@@ -84,9 +76,6 @@ def iter_estimators_which_can_be_tested():
8476
time_series.base.Forecaster,
8577
)
8678

87-
if PYTORCH_INSTALLED:
88-
ignored = (*ignored, PyTorch2RiverBase)
89-
9079
def can_be_tested(estimator):
9180
return not inspect.isabstract(estimator) and not issubclass(estimator, ignored)
9281

0 commit comments

Comments
 (0)