Fix annotation coordinates.

anevolbap · anevolbap · commit 364a360592ee · 2024-03-05T17:56:42.000-05:00
diff --git a/causalpy/skl_experiments.py b/causalpy/skl_experiments.py
@@ -475,7 +475,7 @@ 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",

Original file line number	Diff line number	Diff line change
`@@ -475,7 +475,7 @@ def plot(self, round_to=None):`
`475`	`475`	`)`
`476`	`476`	`ax.annotate(`
`477`	`477`	`"causal\nimpact",`
`478`		`- xy=(1.05, np.mean([self.y_pred_counterfactual, self.y_pred_treatment[1]])),`
	`478`	`+ xy=(1.05, np.mean([self.y_pred_counterfactual[0], self.y_pred_treatment[1]])),`
`479`	`479`	`xycoords="data",`
`480`	`480`	`xytext=(5, 0),`
`481`	`481`	`textcoords="offset points",`