When changing the data-type to pickle, I tested the demos again and found that the viability_plot doesn't work. I tried rolling back a bit but it seems to have been broken for a while, I hadn't noticed before.
I poked around and I think this is due to the way initial_state is initialized. It makes some assumptions on the grids being one-dimensional, which don't fit in the generalized code (where s_grid is a tuple of grids, and each grid can be a different length).
I made a couple changes in the branch viab_plot (3f0a81c4ed27cdddafaf8db4f26aa31f92cd282e), but there are still indexing problems.
When changing the data-type to pickle, I tested the demos again and found that the
viability_plot
doesn't work. I tried rolling back a bit but it seems to have been broken for a while, I hadn't noticed before.I poked around and I think this is due to the way
initial_state
is initialized. It makes some assumptions on the grids being one-dimensional, which don't fit in the generalized code (wheres_grid
is a tuple of grids, and each grid can be a different length).I made a couple changes in the branch
viab_plot
(3f0a81c4ed27cdddafaf8db4f26aa31f92cd282e), but there are still indexing problems.