From 7e2c43581a76d83076c22fc598e41a2fd77817c8 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Sun, 17 Jan 2021 15:27:47 +0200
Subject: [PATCH] Fix interactive session on virtual environment agent changing
 .bashrc/.profile

---
 clearml_session/interactive_session_task.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/clearml_session/interactive_session_task.py b/clearml_session/interactive_session_task.py
index f806b87..a2fc1f7 100644
--- a/clearml_session/interactive_session_task.py
+++ b/clearml_session/interactive_session_task.py
@@ -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