RobertTLange / gymnax-blines

Baselines for gymnax 🤖
Apache License 2.0
57 stars 13 forks source link

replaced deprecated index_update with at[].set() #4

Closed schroederdewitt closed 2 years ago

schroederdewitt commented 2 years ago

see JAX changelog https://jax.readthedocs.io/en/latest/changelog.html?highlight=0.2.22#jax-0-2-22-oct-12-2021

RobertTLange commented 2 years ago

Thank you, looks good! I merged the PR.