Internal change

PiperOrigin-RevId: 721053969
google · Jan 30, 2025 · f5ed79c · f5ed79c
1 parent df8ca35
commit f5ed79c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/grain/_src/python/dataset/transformations/BUILD b/grain/_src/python/dataset/transformations/BUILD
@@ -39,6 +39,7 @@ py_test(
 
 py_test(
     name = "prefetch_test",
+    timeout = "long",
     srcs = ["prefetch_test.py"],
     shard_count = 20,
     srcs_version = "PY3",