Commit 28a51bd1 authored by Chris Jewell's avatar Chris Jewell
Browse files

Fix bug in next generation matrix calcuation.

parent 0ed8920f
......@@ -25,7 +25,7 @@ def calc_posterior_ngm(samples, covar_data):
par = tf.nest.pack_sequence_as(samples, args)
t = events_.shape[-2] - 1
t = par['seir'].shape[-2] - 1
state = compute_state(
samples["init_state"], par['seir'], model_spec.STOICHIOMETRY
)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment