revert to simpler plot titles

Author: drbenvincent

causalpy/experiments/synthetic_control.py

@@ -266,10 +266,10 @@ def _bayesian_plot(
             self.datapre.index,
             self.datapre_treated.sel(treated_units=primary_unit_name),
             "k.",
-            label=f"Observations ({primary_unit_name})",
+            label="Observations",
         )
         handles.append(h)
-        labels.append(f"Observations ({primary_unit_name})")
+        labels.append("Observations")
 
         # post intervention period
         h_line, h_patch = plot_xY(
@@ -298,7 +298,7 @@ def _bayesian_plot(
         handles.append(h)
         labels.append("Causal impact")
 
-        ax[0].set(title=f"{self._get_score_title(round_to)}\nUnit: {primary_unit_name}")
+        ax[0].set(title=f"{self._get_score_title(round_to)}\nUnit")
 
         # MIDDLE PLOT -----------------------------------------------
         plot_xY(
@@ -323,10 +323,10 @@ def _bayesian_plot(
             alpha=0.25,
             label="Causal impact",
         )
-        ax[1].set(title=f"Causal Impact ({primary_unit_name})")
+        ax[1].set(title="Causal Impact")
 
         # BOTTOM PLOT -----------------------------------------------
-        ax[2].set(title=f"Cumulative Causal Impact ({primary_unit_name})")
+        ax[2].set(title="Cumulative Causal Impact")
         plot_xY(
             self.datapost.index,
             self.post_impact_cumulative.sel(treated_units=primary_unit_name),
@@ -415,7 +415,7 @@ def _ols_plot(
             ls=":",
             c="k",
         )
-        ax[0].set(title=f"{self._get_score_title(round_to)}\nUnit: {primary_unit_name}")
+        ax[0].set(title=f"{self._get_score_title(round_to)}\nUnit")
         # Shaded causal effect - handle different prediction formats
         try:
             # For OLS, predictions might be simple arrays
@@ -451,11 +451,11 @@ def _ols_plot(
             label=counterfactual_label,
         )
         ax[1].axhline(y=0, c="k")
-        ax[1].set(title=f"Causal Impact ({primary_unit_name})")
+        ax[1].set(title="Causal Impact")
 
         ax[2].plot(self.datapost.index, self.post_impact_cumulative, c="k")
         ax[2].axhline(y=0, c="k")
-        ax[2].set(title=f"Cumulative Causal Impact ({primary_unit_name})")
+        ax[2].set(title="Cumulative Causal Impact")
 
         # Shaded causal effect
         ax[1].fill_between(

causalpy/tests/test_multi_unit_sc.py

@@ -246,7 +246,6 @@ def test_multi_unit_plotting(self, multi_unit_sc_data):
             assert (
                 "R²" in title or "R^2" in title
             )  # Score information should be displayed
-            assert "Unit:" in title  # Unit information should be displayed
 
     def test_multi_unit_plot_data(self, multi_unit_sc_data):
         """Test that plot data generation works with multiple treated units."""