flax icon indicating copy to clipboard operation
flax copied to clipboard

Published 20 hours ago • verified publisher icon google

remove `tree_map` deprecation warning

Open chiamp opened this issue 10 months ago • 0 comments

Resolves #3842.

Remove tree_map deprecation warning. This was originally added because CLU was causing a deprecation warning at HEAD, but after fixing it and pushing a new release, this is no longer necessary.

NOTE: this will land after Flax upgrades its minimum Python version to 3.10. See #3842 for more details.

chiamp avatar Apr 09 '24 18:04 chiamp