From 3716fb1ca9c78883e82e4829ca16c7dd14ba1441 Mon Sep 17 00:00:00 2001 From: Amog Kamsetty Date: Thu, 24 Sep 2020 14:14:03 -0700 Subject: [PATCH] [Ray SGD] Fix hotfix test (#11015) --- python/ray/util/sgd/tests/test_torch.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/ray/util/sgd/tests/test_torch.py b/python/ray/util/sgd/tests/test_torch.py index d0d9bbdc0..ca0f34fa3 100644 --- a/python/ray/util/sgd/tests/test_torch.py +++ b/python/ray/util/sgd/tests/test_torch.py @@ -111,6 +111,8 @@ def test_apply_all_workers(ray_start_2_cpus, num_workers, use_local): results = trainer.apply_all_workers(fn) assert all(x == 1 for x in results) + trainer.shutdown() + @pytest.mark.parametrize("num_workers", [1, 2] if dist.is_available() else [1]) @pytest.mark.parametrize("use_local", [True, False])