ClearML/clearml-session
1
0
Fork 0
mirror of https://github.com/clearml/clearml-session synced 2025-03-13 23:28:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix interactive session on virtual environment agent changing .bashrc/.profile

Browse Source
This commit is contained in:
allegroai 2021-01-17 15:27:47 +02:00
parent a4581572c1
commit 7e2c43581a
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
clearml_session/interactive_session_task.py
View File

@ -498,6 +498,13 @@ def setup_ssh_server(hostname, hostnames, param, task):
def setup_user_env(param, task):
env = setup_os_env(param)
# do not change user bash/profile
if os.geteuid() != 0:
if param.get("user_key") and param.get("user_secret"):
env['TRAINS_API_ACCESS_KEY'] = param.get("user_key")
env['TRAINS_API_SECRET_KEY'] = param.get("user_secret")
return env
# create symbolic link to the venv
environment = os.path.expanduser('~/environment')
# noinspection PyBroadException