add dim to y_hat

williambdean · williambdean · commit 367c9220b835 · 2025-06-16T17:05:02.000-04:00
diff --git a/causalpy/pymc_models.py b/causalpy/pymc_models.py
@@ -292,7 +292,7 @@ class WeightedSumFitter(PyMCModel):
     """  # noqa: W605
 
     default_priors = {
-        "y_hat": Prior("Normal", sigma=Prior("HalfNormal", sigma=1)),
+        "y_hat": Prior("Normal", sigma=Prior("HalfNormal", sigma=1), dims="obs_ind"),
     }
 
     def build_model(self, X, y, coords):

Original file line number	Diff line number	Diff line change
`@@ -292,7 +292,7 @@ class WeightedSumFitter(PyMCModel):`
`292`	`292`	`""" # noqa: W605`
`293`	`293`
`294`	`294`	`default_priors = {`
`295`		`- "y_hat": Prior("Normal", sigma=Prior("HalfNormal", sigma=1)),`
	`295`	`+ "y_hat": Prior("Normal", sigma=Prior("HalfNormal", sigma=1), dims="obs_ind"),`
`296`	`296`	`}`
`297`	`297`
`298`	`298`	`def build_model(self, X, y, coords):`