update top banner in docs (#343)

cooks distance error fix banner banner
ploomber · Feb 9, 2024 · bfa2b5e · bfa2b5e
1 parent da5cfcc
commit bfa2b5e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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_))