Update pytest.ini with Jraph

google · Jun 17, 2022 · d3da1ea · d3da1ea
1 parent 62c5a03
commit d3da1ea
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pytest.ini b/pytest.ini
@@ -14,5 +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 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