From c64b6945601bfcd50f44197b143ef6a97a67f471 Mon Sep 17 00:00:00 2001 From: Edward Oakes Date: Thu, 28 May 2020 22:38:05 -0500 Subject: [PATCH] Update RaySGD test to use ray.kill instead of __ray_kill__ (#8662) --- python/ray/util/sgd/tests/test_torch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/ray/util/sgd/tests/test_torch.py b/python/ray/util/sgd/tests/test_torch.py index c7a02df24..a6ea2f244 100644 --- a/python/ray/util/sgd/tests/test_torch.py +++ b/python/ray/util/sgd/tests/test_torch.py @@ -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)