File tree 1 file changed +5
-3
lines changed
1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -207,14 +207,14 @@ def test_marginalized_change_point_model(disaster_model):
207
207
ip = m .initial_point ()
208
208
ip .pop ("switchpoint" )
209
209
ref_logp_fn = m .compile_logp (
210
- [m ["switchpoint" ], m ["disasters_observed" ], m ["disasters_missing " ]]
210
+ [m ["switchpoint" ], m ["disasters_observed" ], m ["disasters_unobserved " ]]
211
211
)
212
212
ref_logp = logsumexp ([ref_logp_fn ({** ip , ** {"switchpoint" : year }}) for year in years ])
213
213
214
214
with pytest .warns (UserWarning , match = "There are multiple dependent variables" ):
215
215
m .marginalize (m ["switchpoint" ])
216
216
217
- logp = m .compile_logp ([m ["disasters_observed" ], m ["disasters_missing " ]])(ip )
217
+ logp = m .compile_logp ([m ["disasters_observed" ], m ["disasters_unobserved " ]])(ip )
218
218
np .testing .assert_almost_equal (logp , ref_logp )
219
219
220
220
@@ -241,7 +241,9 @@ def test_marginalized_change_point_model_sampling(disaster_model):
241
241
before_marg ["late_rate" ].mean (), after_marg ["late_rate" ].mean (), rtol = 1e-2
242
242
)
243
243
np .testing .assert_allclose (
244
- before_marg ["disasters_missing" ].mean (), after_marg ["disasters_missing" ].mean (), rtol = 1e-2
244
+ before_marg ["disasters_unobserved" ].mean (),
245
+ after_marg ["disasters_unobserved" ].mean (),
246
+ rtol = 1e-2 ,
245
247
)
246
248
247
249
You can’t perform that action at this time.
0 commit comments