fix tests

CDCgov · Jun 12, 2024 · 1578815 · 1578815
1 parent c4c4328
commit 1578815
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/model/src/test/test_latent_admissions.py b/model/src/test/test_latent_admissions.py
@@ -22,7 +22,7 @@ def test_admissions_sample():
     np.random.seed(223)
     rt = RtRandomWalkProcess()
     with npro.handlers.seed(rng_seed=np.random.randint(1, 600)):
-        sim_rt, *_ = rt.sample(duration=30)
+        sim_rt, *_ = rt.sample(n_timepoints=30)
 
     gen_int = jnp.array([0.5, 0.1, 0.1, 0.2, 0.1])
     i0 = 10 * jnp.ones_like(gen_int)

diff --git a/model/src/test/test_latent_infections.py b/model/src/test/test_latent_infections.py
@@ -19,7 +19,7 @@ def test_infections_as_deterministic():
     np.random.seed(223)
     rt = RtRandomWalkProcess()
     with npro.handlers.seed(rng_seed=np.random.randint(1, 600)):
-        sim_rt, *_ = rt.sample(duration=30)
+        sim_rt, *_ = rt.sample(n_timepoints=30)
 
     gen_int = jnp.array([0.25, 0.25, 0.25, 0.25])