wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-30 17:07:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Ray SGD] Fix hotfix test (#11015)

Browse Source
This commit is contained in:
Amog Kamsetty
2020-09-24 14:14:03 -07:00
committed by GitHub
parent 07bdf062b9
commit 3716fb1ca9
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/util/sgd/tests/test_torch.py
+2
View File
@@ -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])