Skip to content

Commit 15b4b10

Browse files
committed
Improve how runner parses env var file
1 parent 3af50b1 commit 15b4b10

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

deepspeed/launcher/multinode_runner.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import subprocess
55
import warnings
66
from abc import ABC, abstractmethod
7+
from shlex import quote
78

89
from ..utils import logger
910
from .constants import PDSH_MAX_FAN_OUT, MVAPICH_TMP_HOSTFILE

deepspeed/launcher/runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -378,7 +378,7 @@ def main(args=None):
378378
if os.path.isfile(environ_file):
379379
with open(environ_file, 'r') as fd:
380380
for var in fd.readlines():
381-
key, val = var.split('=')
381+
key, val = var.split('=', maxsplit=1)
382382
runner.add_export(key, val)
383383

384384
cmd = runner.get_cmd(env, active_resources)

0 commit comments

Comments
 (0)