diff --git a/river/base/__init__.py b/river/base/__init__.py index a92e437b32..0aaa521934 100644 --- 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 index 756720490a..9eaedb1dda 100644 --- 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", ]