fetch session_token from client_options when aws_log_in=client_option

Signed-off-by: Chinmay Gadgil <[email protected]>
opensearch-project · Oct 13, 2023 · 2210789 · 2210789
1 parent c1d6505
commit 2210789
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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, "