diff --git a/python/ray/tune/BUILD b/python/ray/tune/BUILD
index 8bfb9ddde..3e6dd4388 100644
--- a/python/ray/tune/BUILD
+++ b/python/ray/tune/BUILD
@@ -143,7 +143,7 @@ py_test(
 
 py_test(
     name = "test_api",
-    size = "medium",
+    size = "large",
     srcs = ["tests/test_api.py"],
     deps = [":tune_lib"],
     tags = ["exclusive"],
diff --git a/python/ray/tune/tests/test_api.py b/python/ray/tune/tests/test_api.py
index e578c5359..5b9dabf42 100644
--- a/python/ray/tune/tests/test_api.py
+++ b/python/ray/tune/tests/test_api.py
@@ -514,10 +514,10 @@ class TrainableFunctionApiTest(unittest.TestCase):
             all(t.status == Trial.TERMINATED for t in analysis.trials))
         self.assertTrue(len(analysis.dataframe()) <= top)
 
-        patience = 10
+        patience = 5
         stopper = EarlyStopping("test", top=top, mode="min", patience=patience)
 
-        analysis = tune.run(train, num_samples=100, stop=stopper)
+        analysis = tune.run(train, num_samples=20, stop=stopper)
         self.assertTrue(
             all(t.status == Trial.TERMINATED for t in analysis.trials))
         self.assertTrue(len(analysis.dataframe()) <= patience)