Merge pull request #306 from anevolbap/issue_259-plot-did_skl

Fix #259: Annotation in `did_skl` notebook plot.

Author: drbenvincent

causalpy/skl_experiments.py

@@ -475,7 +475,10 @@ def plot(self, round_to=None):
         )
         ax.annotate(
             "causal\nimpact",
-            xy=(1.05, np.mean([self.y_pred_counterfactual, self.y_pred_treatment[1]])),
+            xy=(
+                1.05,
+                np.mean([self.y_pred_counterfactual[0], self.y_pred_treatment[1]]),
+            ),
             xycoords="data",
             xytext=(5, 0),
             textcoords="offset points",