Loadbalancing Test issue (#452)

* Limiting number of CPUs in loadbalancing test

* fixes as requested
This commit is contained in:
Richard Liaw
2017-04-11 22:33:58 -07:00
committed by Robert Nishihara
parent fb4525f833
commit c3a2505ffd
+4 -3
View File
@@ -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():