From fc54dc865290006cd63b2ccf6784e50635bcdfda Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Tue, 9 Jun 2020 12:45:27 -0700 Subject: [PATCH] [tune] Make test_api faster (#8844) --- python/ray/tune/BUILD | 2 +- python/ray/tune/tests/test_api.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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)