Fix issue with distributed flag
@@ -191,6 +191,7 @@ class PolyaxonFile(object):
if environment:
cluster[TaskType.WORKER] = environment.n_workers
cluster[TaskType.PS] = environment.n_ps
+ if environment.n_workers != 0 or environment.n_ps != 0:
is_distributed = True
return cluster, is_distributed
