Stuck on an issue?
Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.
And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.
FutureWarning: jax.tree_util.tree_multimap() is deprecated
See original GitHub issueLooks like dm-haiku is still using tree_multimap() which is now deprecated (resulting in annoying “future warning” messages with the latest jax)
/usr/local/lib/python3.7/dist-packages/jax/_src/tree_util.py:189: FutureWarning: jax.tree_util.tree_multimap() is deprecated. Please use jax.tree_util.tree_map() instead as a drop-in replacement.
'instead as a drop-in replacement.', FutureWarning)
Issue Analytics
- State:
- Created a year ago
- Comments:5
Top Results From Across the Web
jax.tree_util.tree_multimap() is deprecated · Issue #456 - GitHub
py:189: FutureWarning: jax.tree_util.tree_multimap() is deprecated. Please use jax.tree_util.tree_map() instead as a drop-in replacement. ' ...
Read more >module 'jax' has no attribute 'tree_multimap' in AlphaFold2 ...
jax.tree_multimap was deprecated in JAX version 0.3.5, and removed in JAX version 0.3.16. You can either change the source to use ...
Read more >ColabFold error : r/bioinformatics - Reddit
As it says, that function has been deprecated, meaning that there is a new way of doing it. The updated function works "dropped...
Read more >jax.tree_util package - JAX documentation - Read the Docs
This module provides a small set of utility functions for working with tree-like data structures, such as nested tuples, lists, and dicts. We...
Read more >TF_JAX_tutorials - Part 10 (Pytrees in JAX) - Kaggle
I hope you all have been enjoying the JAX Tutorials so far. ... from jax.tree_util import register_pytree_node from jax.tree_util import ...
Read more >
Top Related Medium Post
No results found
Top Related StackOverflow Question
No results found
Troubleshoot Live Code
Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free
Top Related Reddit Thread
No results found
Top Related Hackernoon Post
No results found
Top Related Tweet
No results found
Top Related Dev.to Post
No results found
Top Related Hashnode Post
No results found
In the meantime, you can add this at the top of your code to hide all the warnings: 😄
Looks like I don’t have the power to reopen the issue. @tomhennigan ?