Closed lucagrementieri closed 2 years ago
The file jax_dataclasses/_copy_and_mutate.py raises many warnings complaining a deprecated function.
jax_dataclasses/_copy_and_mutate.py
FutureWarning: jax.tree_leaves is deprecated, and will be removed in a future release. Use jax.tree_util.tree_leaves instead.
Thanks! Should be fixed in the latest release; let me know if you're still having issues. :slightly_smiling_face:
The file
jax_dataclasses/_copy_and_mutate.py
raises many warnings complaining a deprecated function.