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"],