From d092b126b6755779707be6e4d5b7f52c272181c1 Mon Sep 17 00:00:00 2001 From: Kai Fricke Date: Tue, 14 Feb 2023 09:00:26 -0800 Subject: [PATCH] [tune] Fix two tests after structure refactor deprecation (#32517) #32486 introduced two test failures after hard-depracting a structure refactor. This PR fixes these two stale imports. Signed-off-by: Kai Fricke --- python/ray/tune/tests/test_legacy_import.py | 2 +- python/ray/tune/tests/test_progress_reporter.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/ray/tune/tests/test_legacy_import.py b/python/ray/tune/tests/test_legacy_import.py index 012874caadec..620cfe2205f5 100644 --- a/python/ray/tune/tests/test_legacy_import.py +++ b/python/ray/tune/tests/test_legacy_import.py @@ -47,7 +47,7 @@ def logging_setup(): def test_import_module_raises_warnings(module): py_cmd = ( f"import pytest\n" - f"with pytest.warns(DeprecationWarning):\n" + f"with pytest.raises(DeprecationWarning):\n" f" import {module}\n" f"\n" ) diff --git a/python/ray/tune/tests/test_progress_reporter.py b/python/ray/tune/tests/test_progress_reporter.py index 61ae854b56d3..33cce827a559 100644 --- a/python/ray/tune/tests/test_progress_reporter.py +++ b/python/ray/tune/tests/test_progress_reporter.py @@ -22,7 +22,7 @@ _max_len, ) from ray.tune.result import AUTO_RESULT_KEYS -from ray.tune.trial import Trial +from ray.tune.experiment.trial import Trial EXPECTED_RESULT_1 = """Result logdir: /foo Number of trials: 5 (1 PENDING, 3 RUNNING, 1 TERMINATED)