Skip to content

Commit a60035e

Browse files
committed
get pymc_models tests to pass
1 parent b7300e7 commit a60035e

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

causalpy/pymc_models.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,9 @@ class PyMCModel(pm.Model):
6969
Inference data...
7070
"""
7171

72-
default_priors: dict[str, Any]
72+
@property
73+
def default_priors(self):
74+
return {}
7375

7476
def __init__(
7577
self,
@@ -248,7 +250,7 @@ class LinearRegression(PyMCModel):
248250

249251
default_priors = {
250252
"beta": Prior("Normal", mu=0, sigma=50, dims="coeffs"),
251-
"y_hat": Prior("Normal", sigma=Prior("HalfNormal", sigma=1)),
253+
"y_hat": Prior("Normal", sigma=Prior("HalfNormal", sigma=1), dims="obs_ind"),
252254
}
253255

254256
def build_model(self, X, y, coords):

0 commit comments

Comments
 (0)