diff --git a/python/paddle/distributed/launch/controllers/collective.py b/python/paddle/distributed/launch/controllers/collective.py index 170f19123e3f1269dea974d5c1ca884d0c9ca8bb..de4db754b72cea3db2a616daa4b7df23f333d9f0 100644 --- a/python/paddle/distributed/launch/controllers/collective.py +++ b/python/paddle/distributed/launch/controllers/collective.py @@ -35,9 +35,9 @@ class CollectiveController(Controller): and self.ctx.args.start_port and self.ctx.args.ips ): - self._build_pod_with_args() + return self._build_pod_with_args() else: - self._build_pod_with_master() + return self._build_pod_with_master() def _build_pod_with_args(self): self.pod.replicas = self.pod_replicas()