From abfdffb3aed9b512a30ff06d8be3e139e23edfc2 Mon Sep 17 00:00:00 2001 From: sbidari Date: Thu, 27 Feb 2025 17:27:55 -0500 Subject: [PATCH] coerce population_size to be an integer --- pipelines/build_pyrenew_model.py | 2 +- pyrenew_hew/pyrenew_wastewater_data.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pipelines/build_pyrenew_model.py b/pipelines/build_pyrenew_model.py index 565c9b13..9ecc26b9 100644 --- a/pipelines/build_pyrenew_model.py +++ b/pipelines/build_pyrenew_model.py @@ -103,7 +103,7 @@ def build_model_from_dir( else None ) - population_size = jnp.array(model_data["state_pop"]) + population_size = jnp.array(model_data["state_pop"]).item() pop_fraction = jnp.array(model_data["pop_fraction"]) diff --git a/pyrenew_hew/pyrenew_wastewater_data.py b/pyrenew_hew/pyrenew_wastewater_data.py index 5df556d7..a7c846bf 100644 --- a/pyrenew_hew/pyrenew_wastewater_data.py +++ b/pyrenew_hew/pyrenew_wastewater_data.py @@ -51,7 +51,7 @@ def site_subpop_spine(self): "site_pop": ( self.population_size - site_indices.get_column("site_pop").sum() - ).tolist(), + ), } ) else: