Remove mention of former compat modules (#1588)

* Remove mention of former `compat` modules * Format test_estimators.py
online-ml · Aug 16, 2024 · b5e6fea · b5e6fea
1 parent d606d7b
commit b5e6fea
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 18 deletions.
diff --git a/river/conftest.py b/river/conftest.py
@@ -5,7 +5,6 @@
 try:
     import sklearn  # noqa: F401
 except ImportError:
-    collect_ignore.append("compat/sklearn.py")
     collect_ignore.append("compat/test_sklearn.py")
 
 try:
@@ -14,11 +13,6 @@
     collect_ignore.append("stream/iter_sql.py")
     collect_ignore.append("stream/test_sql.py")
 
-try:
-    import torch  # noqa: F401
-except ImportError:
-    collect_ignore.append("compat/pytorch.py")
-
 try:
     import vaex  # noqa: F401
 except ImportError:

diff --git a/river/test_estimators.py b/river/test_estimators.py
@@ -5,6 +5,7 @@
 import inspect
 
 import pytest
+from sklearn import linear_model as sk_linear_model
 
 from river import (
     anomaly,
@@ -24,15 +25,6 @@
     preprocessing,
     time_series,
 )
-
-try:
-    from river.compat.pytorch import PyTorch2RiverBase
-
-    PYTORCH_INSTALLED = True
-except ImportError:
-    PYTORCH_INSTALLED = False
-from sklearn import linear_model as sk_linear_model
-
 from river.compat.river_to_sklearn import River2SKLBase
 from river.compat.sklearn_to_river import SKL2RiverBase
 
@@ -84,9 +76,6 @@ def iter_estimators_which_can_be_tested():
         time_series.base.Forecaster,
     )
 
-    if PYTORCH_INSTALLED:
-        ignored = (*ignored, PyTorch2RiverBase)
-
     def can_be_tested(estimator):
         return not inspect.isabstract(estimator) and not issubclass(estimator, ignored)