Skip to content

Commit bcce213

Browse files
ferrinericardoV94
authored andcommitted
rename missing to unobserved in test
1 parent dd04eff commit bcce213

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

pymc_experimental/tests/test_marginal_model.py

+5-3
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)