From 2210789c6b50be9cd8ccb4a7f0bfd98d4001e5a2 Mon Sep 17 00:00:00 2001 From: Chinmay Gadgil Date: Thu, 12 Oct 2023 17:51:33 -0700 Subject: [PATCH] fetch session_token from client_options when aws_log_in=client_option Signed-off-by: Chinmay Gadgil --- osbenchmark/client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/osbenchmark/client.py b/osbenchmark/client.py index 21e535fe8..f779a5999 100644 --- a/osbenchmark/client.py +++ b/osbenchmark/client.py @@ -263,7 +263,7 @@ def parse_aws_log_in_params(self): aws_log_in_dict["region"] = self.client_options.get("region") aws_log_in_dict["service"] = self.client_options.get("service") # optional: applicable only for role-based access - aws_log_in_dict["aws_session_token"] = os.environ.get("aws_session_token") + aws_log_in_dict["aws_session_token"] = self.client_options.get("aws_session_token") if (not aws_log_in_dict["aws_access_key_id"] or not aws_log_in_dict["aws_secret_access_key"] or not aws_log_in_dict["service"] or not aws_log_in_dict["region"]): self.logger.error("Invalid amazon aws log in parameters, required input aws_access_key_id, "