Warning Message using Scan

numpyro
1

It seems to me when I run scan. I constantly get a JAX warning message like this. Is this something expected?

FutureWarning: jax.tree_flatten is deprecated, and will be removed in a future release. Use jax.tree_util.tree_flatten instead.
  length = tree_flatten(xs)[0][0].shape[0]
2

i believe this has been fixed in the master branch; no need to be concerned