Change deprecated jax type - Githubissues

adaptive-intelligent-robotics / QDax

Accelerated Quality-Diversity

https://qdax.readthedocs.io/en/latest/

MIT License

258 stars 42 forks source link

Change deprecated jax type #175

Closed LisaCoiffard closed 6 months ago

LisaCoiffard commented 6 months ago

Related issues: N/A

Fix a bug in types.py script, jax.random.KeyArray is deprecated in JAX and should be replaced by jax.Array.

Checks

[x] a clear description of the PR has been added
[ ] sufficient tests have been written
[ ] relevant section added to the documentation
[ ] example notebook added to the repo
[ ] clean docstrings and comments have been written
[ ] if any issue/observation has been discovered, a new issue has been opened

Future improvements

n/a