Commit 37b30227 authored by Chris Jewell's avatar Chris Jewell
Browse files

Add numpy conversion in `predict.py`

parent 28a51bd1
......@@ -77,7 +77,7 @@ def predict(data, posterior_samples, output_file, initial_step, num_steps):
)
prediction = xarray.DataArray(
predicted_events,
predicted_events.numpy(),
coords=[
np.arange(predicted_events.shape[0]),
covar_data.coords["location"],
......@@ -87,7 +87,7 @@ def predict(data, posterior_samples, output_file, initial_step, num_steps):
dims=("iteration", "location", "time", "event"),
)
estimated_init_state = xarray.DataArray(
estimated_init_state,
estimated_init_state.numpy(),
coords=[
np.arange(estimated_init_state.shape[0]),
covar_data.coords["location"],
......
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