mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 13:26:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Don't set PYTHONWARNINGS=error for tests that use TensorFlow.

Browse Source 
Protobuf, which is used by TF, sometimes emits a warning under Python 3.12.

PiperOrigin-RevId: 729554740
This commit is contained in:
Peter Hawkins 2025-02-21 09:22:21 -08:00 committed by jax authors
parent ed10003adc
commit 673a02d614
1 changed files with 3 additions and 0 deletions

3
tests/BUILD

@ -90,6 +90,9 @@ jax_multiplatform_test(
enable_configs = [
"gpu_p100x2",
],
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"),
)