From bfa2b5e1b0a143ebb0950982b33ba41e9397f6a2 Mon Sep 17 00:00:00 2001
From: Neelasha Sen <neelasha.sen@gmail.com>
Date: Fri, 9 Feb 2024 20:12:58 +0530
Subject: [PATCH] update top banner in docs (#343)

cooks distance error fix

banner

banner
---
 doc/conf.py                               | 6 ++++--
 src/sklearn_evaluation/plot/regression.py | 4 +---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/doc/conf.py b/doc/conf.py
index e51b733e..2d5c3420 100644
--- a/doc/conf.py
+++ b/doc/conf.py
@@ -49,8 +49,10 @@
 html_theme_options = {
     "analytics": {"google_analytics_id": "G-3KL9PSJBZZ"},
     "announcement": (
-        "Host Voilà and Streamlit apps for free with "
-        "<a href='https://www.platform.ploomber.io/register/'>Ploomber Cloud!</a>"
+        "Deploy AI apps for free on "
+        "<a href='https://ploomber.io/?utm_medium=readthedocs&utm_source=sklearn-evaluation' "  # noqa
+        "style='color:#7CB9E8;' >"
+        "Ploomber Cloud!</a>"
     ),
     "search_bar_text": "Search this book...",
     "launch_buttons": {
diff --git a/src/sklearn_evaluation/plot/regression.py b/src/sklearn_evaluation/plot/regression.py
index b038176d..ccda6ce9 100644
--- a/src/sklearn_evaluation/plot/regression.py
+++ b/src/sklearn_evaluation/plot/regression.py
@@ -204,9 +204,7 @@ def cooks_distance(X, y, ax=None):
 
     if ax is None:
         _, ax = plt.subplots()
-    _, _, baseline = ax.stem(
-        distance_, linefmt="C0-", markerfmt=",", use_line_collection=True
-    )
+    _, _, baseline = ax.stem(distance_, linefmt="C0-", markerfmt=",")
 
     ax.set_xlim(0, len(distance_))