Simplify logic with dict.get() (#423)

dmlc · Jul 16, 2018 · 951835b · 951835b
1 parent 99696ad
commit 951835b
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tracker/dmlc_tracker/local.py b/tracker/dmlc_tracker/local.py
@@ -22,9 +22,7 @@ def exec_cmd(cmd, role, taskid, pass_env):
     env['DMLC_ROLE'] = role
     env['DMLC_JOB_CLUSTER'] = 'local'
 
-    num_retry = 0
-    if 'DMLC_NUM_ATTEMPT' in env:
-        num_retry = env['DMLC_NUM_ATTEMPT']
+    num_retry = env.get('DMLC_NUM_ATTEMPT', 0)
 
     while True:
         if os.name == 'nt':