From 1e48a033af1cbc7297cc9b129a3f0eb247239c5a Mon Sep 17 00:00:00 2001 From: matthewdeng Date: Wed, 18 Sep 2024 18:43:28 -0700 Subject: [PATCH] [train] update quick start timeout (#47732) The time is spent mostly on downloading the dataset to disk, and training times out after 15 seconds. Bumping to 300 seconds. Signed-off-by: Matthew Deng Co-authored-by: Hongpeng Guo --- python/ray/train/BUILD | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/train/BUILD b/python/ray/train/BUILD index 8a566348a922..c5168c17ab12 100644 --- a/python/ray/train/BUILD +++ b/python/ray/train/BUILD @@ -118,7 +118,7 @@ py_test( py_test( name = "torch_quick_start", - size = "small", + size = "medium", main = "examples/pytorch/torch_quick_start.py", srcs = ["examples/pytorch/torch_quick_start.py"], tags = ["team:ml", "exclusive"],