Replace the deprecated index_update

pyro-ppl / funsor

Functional tensors for probabilistic programming

https://funsor.pyro.ai

Apache License 2.0

236 stars 20 forks source link

Closed fehiepsi closed 3 years ago

fehiepsi commented 3 years ago

index_update is deprecated in favor of src.at(indices).set(desc). See its docs

fritzo commented 3 years ago

do we need to bump the jax version in this PR?

fehiepsi commented 3 years ago

Yes, the minimal version in setup.py does not have this feature yet. So I just bumped the version up a bit.