[LAUNCH] fix set args bug (#44717)

python/paddle/distributed/launch/context/__init__.py

@@ -101,7 +101,6 @@ class Context(object):
             return False
 
     def set_env_in_args(self):
-        # this logic may not propre to replace args with env, but ...
         for k, v in env_args_mapping.items():
             if k in self.envs:
-                setattr(self.args, v, type(getattr(self.args, v))(self.envs[k]))
+                setattr(self.args, v, self.envs[k])

python/paddle/distributed/launch/controllers/collective.py

@@ -131,7 +131,8 @@ class CollectiveElasticController(CollectiveController):
 
     def run(self):
 
-        timeout = self.ctx.args.elastic_timeout if self.job.elastic else self.ctx.args.elastic_timeout * 10
+        timeout = int(self.ctx.args.elastic_timeout)
+        timeout = timeout if self.job.elastic else timeout * 10
         self.register()
 
         while self.pod.restart <= self.ctx.args.max_restart:

python/paddle/distributed/launch/plugins/__init__.py

@@ -62,7 +62,7 @@ def rewrite_host_ip(ctx):
 
 
 def test_mode(ctx):
-    if ctx.args.training_script == 'test':
+    if ctx.args.training_script == 'run_check':
         ctx.logger.info('Paddle Distributed Test begin...')
         if int(ctx.args.nnodes) < 2:
             ctx.args.nnodes = 2