From dcb0f98e9eabe4d022e0c96684496d247cf2339c Mon Sep 17 00:00:00 2001
From: Mo3ad-S <155067453+Mo3ad-S@users.noreply.github.com>
Date: Tue, 26 Nov 2024 19:49:45 +0100
Subject: [PATCH] updated the rest of the project

---
 river/base/__init__.py         | 2 --
 river/linear_model/__init__.py | 2 ++
 2 files changed, 2 insertions(+), 2 deletions(-)

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",
 ]