fix test

online-ml · May 31, 2024 · 828d7f5 · 828d7f5
1 parent 6dfba3b
commit 828d7f5
Show file tree

Hide file tree

Showing 2 changed files with 597 additions and 0 deletions.
diff --git a/river/model_selection/__init__.py b/river/model_selection/__init__.py
@@ -18,12 +18,14 @@
 from .bandit import BanditClassifier, BanditRegressor
 from .greedy import GreedyRegressor
 from .sh import SuccessiveHalvingClassifier, SuccessiveHalvingRegressor
+from .sspt import SSPT
 
 __all__ = [
     "base",
     "BanditClassifier",
     "BanditRegressor",
     "GreedyRegressor",
+    "SSPT",
     "SuccessiveHalvingClassifier",
     "SuccessiveHalvingRegressor",
 ]