From 90627e6d91cc4a78fd94765d97d0e307f219dad7 Mon Sep 17 00:00:00 2001 From: "George G. Vega Yon" Date: Wed, 3 Apr 2024 13:30:06 -0600 Subject: [PATCH] Different vector for hosp_report_prob_dist in tests --- model/src/test/test_model_hospitalizations.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/model/src/test/test_model_hospitalizations.py b/model/src/test/test_model_hospitalizations.py index c9ab229e..3cc2d1d2 100644 --- a/model/src/test/test_model_hospitalizations.py +++ b/model/src/test/test_model_hospitalizations.py @@ -356,13 +356,22 @@ def test_model_hosp_with_obs_model_weekday_phosp(): # Other random components weekday = jnp.array([1, 1, 1, 1, 2, 2]) - weekday = weekday / weekday.sum() weekday = jnp.tile(weekday, 10) + weekday = weekday / weekday.sum() weekday = weekday[:31] - hosp_report_prob_dist = DeterministicVariable((weekday,)) weekday = DeterministicVariable((weekday,)) + hosp_report_prob_dist = jnp.array([0.9, 0.8, 0.7, 0.7, 0.6, 0.4]) + hosp_report_prob_dist = jnp.tile(hosp_report_prob_dist, 10) + hosp_report_prob_dist = hosp_report_prob_dist / hosp_report_prob_dist.sum() + + hosp_report_prob_dist = hosp_report_prob_dist[:31] + + hosp_report_prob_dist = DeterministicVariable( + vars=(hosp_report_prob_dist,) + ) + latent_hospitalizations = HospitalAdmissions( infection_to_admission_interval=inf_hosp, infections_varname="infections",