resolving conflicts

JeanVanDyk · JeanVanDyk · commit 701fe1396e4f · 2025-09-24T15:49:04.000+02:00
diff --git a/causalpy/pymc_models.py b/causalpy/pymc_models.py
@@ -1035,11 +1035,12 @@ def _data_setter(self, X) -> None:
             treated_units_coord = getattr(self, "coords", {}).get(
                 "treated_units", ["unit_0"]
             )
+            n_treated_unit = len(treated_units_coord)
             pm.set_data(
                 {
                     "X": X,
                     "t": np.arange(n_obs)[:, None],
-                    "y": np.zeros((n_obs, treated_units_coord)),
+                    "y": np.zeros((n_obs, n_treated_unit)),
                 },
                 coords={"obs_ind": np.arange(n_obs)},
             )

Original file line number	Diff line number	Diff line change
`@@ -1035,11 +1035,12 @@ def _data_setter(self, X) -> None:`
`1035`	`1035`	`treated_units_coord = getattr(self, "coords", {}).get(`
`1036`	`1036`	`"treated_units", ["unit_0"]`
`1037`	`1037`	`)`
	`1038`	`+ n_treated_unit = len(treated_units_coord)`
`1038`	`1039`	`pm.set_data(`
`1039`	`1040`	`{`
`1040`	`1041`	`"X": X,`
`1041`	`1042`	`"t": np.arange(n_obs)[:, None],`
`1042`		`- "y": np.zeros((n_obs, treated_units_coord)),`
	`1043`	`+ "y": np.zeros((n_obs, n_treated_unit)),`
`1043`	`1044`	`},`
`1044`	`1045`	`coords={"obs_ind": np.arange(n_obs)},`
`1045`	`1046`	`)`