wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-04 13:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Start moving ray internal files to _private module (#10994)

Browse Source
This commit is contained in:
Eric Liang
2020-09-24 22:46:35 -07:00
committed by GitHub
parent 3b6fe72029
commit 609c1b8acd
39 changed files with 1825 additions and 1831 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/tests/test_sync.py
+3 -2
View File
@@ -113,7 +113,8 @@ class TestSyncFunctionality(unittest.TestCase):
}).trials
with patch.object(CommandBasedClient, "_execute") as mock_fn:
with patch("ray.services.get_node_ip_address") as mock_sync:
with patch(
"ray._private.services.get_node_ip_address") as mock_sync:
sync_config = tune.SyncConfig(
sync_to_driver="echo {source} {target}")
mock_sync.return_value = "0.0.0.0"
@@ -209,7 +210,7 @@ class TestSyncFunctionality(unittest.TestCase):
test_file_path = os.path.join(trial.logdir, "test.log2")
self.assertFalse(os.path.exists(test_file_path))
with patch("ray.services.get_node_ip_address") as mock_sync:
with patch("ray._private.services.get_node_ip_address") as mock_sync:
mock_sync.return_value = "0.0.0.0"
sync_config = tune.SyncConfig(sync_to_driver=sync_func_driver)
[trial] = tune.run(