From d03bbc0d0fc1b497a53e7706c7c072bdf965435f Mon Sep 17 00:00:00 2001
From: jax authors <no-reply@google.com>
Date: Tue, 17 Oct 2023 12:58:39 -0700
Subject: [PATCH] random_lax_test: Bump shards number for CPU config.

PiperOrigin-RevId: 574239793
---
 tests/BUILD | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/BUILD b/tests/BUILD
index 9b50e0c78..4cbdb3bcc 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -731,7 +731,7 @@ jax_test(
         ],
     },
     shard_count = {
-        "cpu": 30,
+        "cpu": 40,
         "gpu": 30,
         "tpu": 40,
         "iree": 30,