ClearML/clearml-session
1
0
Fork 0
mirror of https://github.com/clearml/clearml-session synced 2025-01-31 02:46:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix env variables inherited by SSH

Browse Source
This commit is contained in:
allegroai 2023-02-11 23:14:44 +02:00
parent d961c4b83d
commit 2ce7e3fabd
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clearml_session/interactive_session_task.py
View File

@ -460,6 +460,12 @@ def setup_ssh_server(hostname, hostnames, param, task, env):
if not param.get("ssh_server"):
return
# make sure we do not pass it along to others, work on a copy
env = deepcopy(env)
env.pop('LOCAL_PYTHON', None)
env.pop('PYTHONPATH', None)
env.pop('DEBIAN_FRONTEND', None)
print("Installing SSH Server on {} [{}]".format(hostname, hostnames))
ssh_password = param.get("ssh_password", "training")
# noinspection PyBroadException
@ -601,8 +607,7 @@ def setup_ssh_server(hostname, hostnames, param, task, env):
except Exception:
pass
proc_args = [sshd_path, "dropbear", "-e", "-K", "30", "-I", "0", "-F", "-p", str(port)] + dropbear_key_files
# make sure we do not pass it along to others, work on a copy
env = deepcopy(env)
# this is a copy oof `env` so there is nothing to worry about
env["DROPBEAR_CLEARML_FIXED_PASSWORD"] = ssh_password
else:
proc_args = [sshd_path, "-D", "-p", str(port)] + (["-f", custom_ssh_conf] if custom_ssh_conf else [])