[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 <coding@kaifricke.com>
ray-project · Feb 14, 2023 · d092b12 · d092b12
1 parent 66c0533
commit d092b12
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)