wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 02:01:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update RaySGD test to use ray.kill instead of __ray_kill__ (#8662)

Browse Source
This commit is contained in:
Edward Oakes
2020-05-28 22:38:05 -05:00
committed by GitHub
parent 76450c8d47
commit c64b694560
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/util/sgd/tests/test_torch.py
+2 -2
View File
@@ -649,7 +649,7 @@ def test_resize(ray_start_2_cpus): # noqa: F811
if self._num_failures < 1:
time.sleep(1) # Make the batch will fail correctly.
self.remote_workers[0].__ray_kill__()
ray.kill(self.remote_workers[0])
try:
local_worker_stats = self.local_worker.train_epoch(**params)
@@ -698,7 +698,7 @@ def test_fail_twice(ray_start_2_cpus): # noqa: F811
if self._num_failures < 2:
time.sleep(1) # Make the batch will fail correctly.
self.remote_workers[0].__ray_kill__()
ray.kill(self.remote_workers[0])
try:
local_worker_stats = self.local_worker.train_epoch(**params)