From 3b2e4a16007e4554fe7e1aece87b0f0968d77676 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Thu, 7 Nov 2024 14:11:45 -0800
Subject: [PATCH] Remove sharding from custom_root_test.

This test only takes around 30s on most hardware platforms, it does not need 10 shards.

PiperOrigin-RevId: 694243316
---
 tests/BUILD | 5 -----
 1 file changed, 5 deletions(-)

diff --git a/tests/BUILD b/tests/BUILD
index 90a8913a1825..ba725842b36c 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -416,11 +416,6 @@ jax_multiplatform_test(
 jax_multiplatform_test(
     name = "custom_root_test",
     srcs = ["custom_root_test.py"],
-    shard_count = {
-        "cpu": 10,
-        "gpu": 10,
-        "tpu": 10,
-    },
 )
 
 jax_multiplatform_test(