Ignore the Deprecation warning produced about `native_serialization=F…

…alse`. PiperOrigin-RevId: 657221363
jax-ml · Jul 29, 2024 · 6127baa · 6127baa
1 parent fd23b87
commit 6127baa
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/jax_to_ir_test.py b/tests/jax_to_ir_test.py
@@ -81,6 +81,10 @@ def test_parse_shape_str_invalid(self):
       jax_to_ir.parse_shape_str('foo[]')
 
   @unittest.skipIf(tf is None, 'TensorFlow not installed.')
+  @jtu.ignore_warning(
+      category=UserWarning,
+      message='jax2tf.convert with native_serialization=False is deprecated.'
+  )
   def test_jax_to_tf_axpy(self):
     tf_proto, tf_text = jax_to_ir.jax_to_tf(axpy, [
         ('y', jax_to_ir.parse_shape_str('f32[128]')),