File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
Expand file tree Collapse file tree 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):
207207 ip = m .initial_point ()
208208 ip .pop ("switchpoint" )
209209 ref_logp_fn = m .compile_logp (
210- [m ["switchpoint" ], m ["disasters_observed" ], m ["disasters_missing " ]]
210+ [m ["switchpoint" ], m ["disasters_observed" ], m ["disasters_unobserved " ]]
211211 )
212212 ref_logp = logsumexp ([ref_logp_fn ({** ip , ** {"switchpoint" : year }}) for year in years ])
213213
214214 with pytest .warns (UserWarning , match = "There are multiple dependent variables" ):
215215 m .marginalize (m ["switchpoint" ])
216216
217- logp = m .compile_logp ([m ["disasters_observed" ], m ["disasters_missing " ]])(ip )
217+ logp = m .compile_logp ([m ["disasters_observed" ], m ["disasters_unobserved " ]])(ip )
218218 np .testing .assert_almost_equal (logp , ref_logp )
219219
220220
@@ -241,7 +241,9 @@ def test_marginalized_change_point_model_sampling(disaster_model):
241241 before_marg ["late_rate" ].mean (), after_marg ["late_rate" ].mean (), rtol = 1e-2
242242 )
243243 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 ,
245247 )
246248
247249
You can’t perform that action at this time.
0 commit comments