Disable tests that are flaky in CI.

PiperOrigin-RevId: 547464866
This commit is contained in:
Peter Hawkins 2023-07-12 05:13:03 -07:00 committed by jax authors
parent f416aaa66b
commit 3fcf72af32

View File

@ -639,6 +639,7 @@ jax_test(
"tpu": [
"optonly",
"nomsan", # Times out
"notsan", # Times out
],
},
shard_count = {