@@ -25,7 +25,7 @@ def model_nc():
25
25
return mod
26
26
27
27
28
- @pytest .mark .parameterize ("var" , ["g" , "e" ])
28
+ @pytest .mark .parametrize ("var" , ["g" , "e" ])
29
29
def test_reparametrize_created (model_c : pm .Model , var ):
30
30
model_reparam , vip = vip_reparametrize (model_c , [var ])
31
31
assert f"{ var } " in vip .get_lambda ()
@@ -35,7 +35,7 @@ def test_reparametrize_created(model_c: pm.Model, var):
35
35
assert ~ np .isfinite (model_reparam [f"{ var } ::lam_logit__" ].get_value ()).any ()
36
36
37
37
38
- @pytest .mark .parameterize ("var" , ["g" , "e" ])
38
+ @pytest .mark .parametrize ("var" , ["g" , "e" ])
39
39
def test_random_draw (model_c : pm .Model , model_nc , var ):
40
40
model_c = pm .do (model_c , {"m" : 3 , "s" : 2 })
41
41
model_nc = pm .do (model_nc , {"m" : 3 , "s" : 2 })
@@ -60,7 +60,7 @@ def test_random_draw(model_c: pm.Model, model_nc, var):
60
60
np .testing .assert_allclose (v_1 .std (), nc .std ())
61
61
62
62
63
- @pytest .mark .parameterize ("var" , ["g" , "e" ])
63
+ @pytest .mark .parametrize ("var" , ["g" , "e" ])
64
64
def test_reparam_fit (model_c , var ):
65
65
model_v , vip = vip_reparametrize (model_c , [var ])
66
66
with model_v :
0 commit comments