Addressing pending comments

CDCgov · May 8, 2024 · f71bcda · f71bcda
1 parent cc30346
commit f71bcda
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/model/pyproject.toml b/model/pyproject.toml
@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "pyrenew"
 version = "0.1.0"
-description = "Pyrenew: a package for Bayesian renewal modeling with JAX and Numpyro\""
+description = "Pyrenew: a package for Bayesian renewal modeling with JAX and Numpyro"
 authors = ["Dylan H. Morris <[email protected]>"]
 license = "Apache 2.0"
 readme = "README.md"
@@ -16,7 +16,6 @@ jax = "^0.4.24"
 numpy = "^1.26.4"
 polars = "^0.20.13"
 pillow = "^10.3.0" # See #56 on CDCgov/multisignal-epi-inference
-numpydoc = "^1.7.0"
 
 [tool.poetry.group.dev]
 optional = true
@@ -29,6 +28,7 @@ nbclient = "^0.10.0"
 nbformat = "^5.10.0"
 pytest-cov = "^5.0.0"
 pytest-mpl = "^0.17.0"
+numpydoc = "^1.7.0"
 
 [tool.numpydoc_validation]
 checks = [

diff --git a/model/src/pyrenew/datasets/wastewater.py b/model/src/pyrenew/datasets/wastewater.py
@@ -29,7 +29,7 @@ def load_wastewater() -> pl.DataFrame:
         - `log_conc`
         - `date`
         - `load_sewage`
-        - `below_LOAD`
+        - `below_load`
         - `daily_hosp_admits`
         - `daily_hosp_admits_for_eval`
         - `pop`

diff --git a/model/src/pyrenew/datasets/wastewater.tsv b/model/src/pyrenew/datasets/wastewater.tsv
@@ -1,4 +1,4 @@
-t	lab_wwtp_unique_id	log_conc	date	load_sewage	below_LOAD	daily_hosp_admits	daily_hosp_admits_for_eval	pop	forecast_date	hosp_calibration_time	site	ww_pop	inf_per_capita
+t	lab_wwtp_unique_id	log_conc	date	load_sewage	below_load	daily_hosp_admits	daily_hosp_admits_for_eval	pop	forecast_date	hosp_calibration_time	site	ww_pop	inf_per_capita
 1	1		2023-10-30			6	6	1e+06	2024-02-05	90	1	4e+05	0.000663426572737469
 1	2		2023-10-30			6	6	1e+06	2024-02-05	90	1	4e+05	0.000663426572737469
 1	3		2023-10-30			6	6	1e+06	2024-02-05	90	2	2e+05	0.000663426572737469