From c3a2505ffd42de8bf49f9bb99cdc0e496fc5fcea Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Tue, 11 Apr 2017 22:33:58 -0700 Subject: [PATCH] Loadbalancing Test issue (#452) * Limiting number of CPUs in loadbalancing test * fixes as requested --- test/runtest.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/test/runtest.py b/test/runtest.py index 888500386..a84e26c1f 100644 --- a/test/runtest.py +++ b/test/runtest.py @@ -1328,10 +1328,11 @@ class SchedulingAlgorithm(unittest.TestCase): def testLoadBalancing(self): # This test ensures that tasks are being assigned to all local schedulers # in a roughly equal manner. - num_workers = 21 num_local_schedulers = 3 - ray.worker._init(start_ray_local=True, num_workers=num_workers, - num_local_schedulers=num_local_schedulers) + num_cpus = 7 + ray.worker._init(start_ray_local=True, + num_local_schedulers=num_local_schedulers, + num_cpus=num_cpus) @ray.remote def f():