Merge pull request #2209 from marcvanzee:pytest-tree-multimap

PiperOrigin-RevId: 456236099
google · Jun 21, 2022 · 50a5444 · 50a5444
2 parents 4042eee + d3da1ea
commit 50a5444
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytest.ini b/pytest.ini
@@ -14,7 +14,7 @@ filterwarnings =
     ignore:can't resolve package from.*:ImportWarning
 # Jax imports flatbuffers which imports imp in a compat file.
     ignore:the imp module is deprecated.*:DeprecationWarning
-# TODO(#2037): Remove this once Optax releases a new version.
+# TODO(#2037): Remove this once Jgraph uses jax.tree_map.
     ignore:jax.tree_util.tree_multimap\(\) is deprecated.*:FutureWarning
 # traverse_util.Traversal will be removed soon.
     ignore:`flax.traverse_util.Traversal` will be deprecated.*:DeprecationWarning