ClearML/clearml-session
1
0
Fork 0
mirror of https://github.com/clearml/clearml-session synced 2025-06-26 18:16:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix interactive nonroot set .profile

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

22
clearml_session/interactive_session_task.py
View File

@ -683,12 +683,24 @@ def setup_user_env(param, task):
except Exception as ex: except Exception as ex:
print('Applying vault configuration failed: {}'.format(ex)) print('Applying vault configuration failed: {}'.format(ex))
# do not change user bash/profile # do not change user bash/profile if we are not running inside a container
if os.geteuid() != 0: if os.geteuid() != 0:
if param.get("user_key") and param.get("user_secret"): # check if we are inside a container
env['CLEARML_API_ACCESS_KEY'] = param.get("user_key") is_container = False
env['CLEARML_API_SECRET_KEY'] = param.get("user_secret") try:
return env with open("/proc/1/sched", "rt") as f:
lines = f.readlines()
if lines and lines[0].split()[0] in ("bash", "sh", "zsh"):
# this a container
is_container = True
except Exception: # noqa
pass
if not is_container:
if param.get("user_key") and param.get("user_secret"):
env['CLEARML_API_ACCESS_KEY'] = param.get("user_key")
env['CLEARML_API_SECRET_KEY'] = param.get("user_secret")
return env
# create symbolic link to the venv # create symbolic link to the venv
environment = os.path.expanduser('~/environment') environment = os.path.expanduser('~/environment')