updated the rest of the project

online-ml · Nov 26, 2024 · dcb0f98 · dcb0f98
1 parent 322b924
commit dcb0f98
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/river/base/__init__.py b/river/base/__init__.py
@@ -15,7 +15,6 @@
 from __future__ import annotations
 
 from . import tags, typing
-from .adpredictor import AdPredictor
 from .base import Base
 from .classifier import Classifier, MiniBatchClassifier
 from .clusterer import Clusterer
@@ -38,7 +37,6 @@
 from .wrapper import Wrapper
 
 __all__ = [
-    "AdPredictor",
     "Base",
     "BinaryDriftDetector",
     "BinaryDriftAndWarningDetector",

diff --git a/river/linear_model/__init__.py b/river/linear_model/__init__.py
@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from . import base
+from .adpredictor import AdPredictor
 from .alma import ALMAClassifier
 from .bayesian_lin_reg import BayesianLinearRegression
 from .lin_reg import LinearRegression
@@ -21,4 +22,5 @@
     "PARegressor",
     "Perceptron",
     "SoftmaxRegression",
+    "AdPredictor",
 ]