Deleting Warning on Plot

pymc-labs · Sep 5, 2023 · 7305590 · 7305590
1 parent 05d6bb5
commit 7305590
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pymc_marketing/mmm/base.py b/pymc_marketing/mmm/base.py
@@ -832,7 +832,6 @@ def _plot_response_curve_fit(
         ax.plot(
             x_inflection,
             y_inflection,
-            "go",
             color=f"C{color_index}",
             markerfacecolor="white",
         )
@@ -983,7 +982,7 @@ def plot_direct_contribution_curves(
                 x = self.X[self.channel_columns].to_numpy()[:, i]
                 y = channel_contributions.sel(channel=channel).to_numpy()
 
-                ax.scatter(x, y, label=f"{channel}", color=f"C{i}")
+                ax.scatter(x, y, label="Data Points", color=f"C{i}")
 
                 if show_fit:
                     self._plot_response_curve_fit(