Fix warnings in array_interoperability_test.

PiperOrigin-RevId: 747586780

Author: hawkinsp

tests/BUILD

@@ -102,9 +102,6 @@ jax_multiplatform_test(
     enable_configs = [
         "gpu_h100x2",
     ],
-    env = {
-        "PYTHONWARNINGS": "default",  # TODO(b/394123878): protobuf, via TensorFlow, issues a Python warning under Python 3.12+ sometimes.
-    },
     tags = ["multiaccelerator"],
     deps = py_deps("tensorflow_core"),
 )

tests/array_interoperability_test.py

@@ -95,6 +95,10 @@ def setUp(self):
       message="Calling from_dlpack with a DLPack tensor",
       category=DeprecationWarning,
   )
+  @jtu.ignore_warning(
+      message="jax.dlpack.to_dlpack was deprecated.*",
+      category=DeprecationWarning,
+  )
   def testJaxRoundTrip(self, shape, dtype, copy, use_stream):
     rng = jtu.rand_default(self.rng())
     np = rng(shape, dtype)
@@ -188,6 +192,10 @@ def testTensorFlowToJax(self, shape, dtype):
     dtype=dlpack_dtypes,
   )
   @unittest.skipIf(not tf, "Test requires TensorFlow")
+  @jtu.ignore_warning(
+      message="jax.dlpack.to_dlpack was deprecated.*",
+      category=DeprecationWarning,
+  )
   def testJaxToTensorFlow(self, shape, dtype):
     if (not config.enable_x64.value and
         dtype in [jnp.int64, jnp.uint64, jnp.float64]):