File tree 2 files changed +1
-18
lines changed
2 files changed +1
-18
lines changed Original file line number Diff line number Diff line change 5
5
try :
6
6
import sklearn # noqa: F401
7
7
except ImportError :
8
- collect_ignore .append ("compat/sklearn.py" )
9
8
collect_ignore .append ("compat/test_sklearn.py" )
10
9
11
10
try :
14
13
collect_ignore .append ("stream/iter_sql.py" )
15
14
collect_ignore .append ("stream/test_sql.py" )
16
15
17
- try :
18
- import torch # noqa: F401
19
- except ImportError :
20
- collect_ignore .append ("compat/pytorch.py" )
21
-
22
16
try :
23
17
import vaex # noqa: F401
24
18
except ImportError :
Original file line number Diff line number Diff line change 5
5
import inspect
6
6
7
7
import pytest
8
+ from sklearn import linear_model as sk_linear_model
8
9
9
10
from river import (
10
11
anomaly ,
24
25
preprocessing ,
25
26
time_series ,
26
27
)
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
-
36
28
from river .compat .river_to_sklearn import River2SKLBase
37
29
from river .compat .sklearn_to_river import SKL2RiverBase
38
30
@@ -84,9 +76,6 @@ def iter_estimators_which_can_be_tested():
84
76
time_series .base .Forecaster ,
85
77
)
86
78
87
- if PYTORCH_INSTALLED :
88
- ignored = (* ignored , PyTorch2RiverBase )
89
-
90
79
def can_be_tested (estimator ):
91
80
return not inspect .isabstract (estimator ) and not issubclass (estimator , ignored )
92
81
You can’t perform that action at this time.
0 commit comments