Added DMLC_INTERFACE env var passing to ssh launcher (apache#297)

* Added DMLC_INTERFACE env var passing to ssh launcher * Shorter line
eric-haibin-lin · Aug 24, 2017 · 87a0c16 · 87a0c16
1 parent e880afe
commit 87a0c16
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tracker/dmlc_tracker/ssh.py b/tracker/dmlc_tracker/ssh.py
@@ -23,7 +23,8 @@ def sync_dir(local_dir, slave_node, slave_dir):
 def get_env(pass_envs):
     envs = []
     # get system envs
-    keys = ['OMP_NUM_THREADS', 'KMP_AFFINITY', 'LD_LIBRARY_PATH', 'AWS_ACCESS_KEY_ID', 'AWS_SECRET_ACCESS_KEY']
+    keys = ['OMP_NUM_THREADS', 'KMP_AFFINITY', 'LD_LIBRARY_PATH', 'AWS_ACCESS_KEY_ID',
+            'AWS_SECRET_ACCESS_KEY', 'DMLC_INTERFACE']
     for k in keys:
         v = os.getenv(k)
         if v is not None: