Remove tree_map
deprecation filter after Flax upgrades minimum Python version to 3.10
#3842
Labels
Priority: P2 - no schedule
Best effort response and resolution. We have no plan to work on this at the moment.
Context:
jax.tree_map
is deprecatedtree_map
#3823 renames alljax.tree_map
usages tojax.tree_util.tree_map
in Flax, however we get an error in CI because of a CLU dependencytree_map
fix) on python versions less than 3.10, since the match-case syntax used by CLU is only available in Python 3.10 or greatertree_map
deprecation warning filter #3828Once Flax upgrades its minimum Python version to 3.10, we should remove the deprecation warning filter and remove enforcing an earlier version of CLU.
The text was updated successfully, but these errors were encountered: