fixed ray bug

This commit is contained in:
Benoît Sierro
2021-09-02 16:07:11 +02:00
parent 4b2495ca6d
commit 838e962d17

View File

@@ -689,8 +689,8 @@ class RaySimulations(Simulations, priority=2):
@property @property
def sim_jobs_total(self): def sim_jobs_total(self):
if self.configuration.config.worker_num is not None: if self.configuration.worker_num is not None:
return self.configuration.config.worker_num return self.configuration.worker_num
tot_cpus = ray.cluster_resources().get("CPU", 1) tot_cpus = ray.cluster_resources().get("CPU", 1)
return int(min(self.configuration.num_sim, tot_cpus)) return int(min(self.configuration.num_sim, tot_cpus))