diff --git a/python/ray/tune/BUILD b/python/ray/tune/BUILD index 41d547720..95e16345a 100644 --- a/python/ray/tune/BUILD +++ b/python/ray/tune/BUILD @@ -74,7 +74,7 @@ py_test( py_test( name = "test_progress_reporter", - size = "small", + size = "medium", srcs = ["tests/test_progress_reporter.py"], deps = [":tune_lib"], ) diff --git a/python/ray/tune/tests/test_progress_reporter.py b/python/ray/tune/tests/test_progress_reporter.py index 83e930153..ee12dc8ff 100644 --- a/python/ray/tune/tests/test_progress_reporter.py +++ b/python/ray/tune/tests/test_progress_reporter.py @@ -212,7 +212,6 @@ class ProgressReporterTest(unittest.TestCase): print(prog2) assert prog2 == EXPECTED_RESULT_2 - @pytest.mark.skip("Very flaky in CI.") def testEndToEndReporting(self): with tempfile.NamedTemporaryFile(suffix=".py") as f: f.write(END_TO_END_COMMAND.encode("utf-8"))