using boto3 session region directly for log configuration region (#60)

AgnostiqHQ · Dec 6, 2022 · 81ab2bc · 81ab2bc
1 parent 4d7c7ec
commit 81ab2bc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [UNRELEASED]
 
+### Changed
+
+- Using region value directly from boto3 session to configure logging to support cases where user supplied region is empty
+
 ## [0.27.0] - 2022-11-28
 
 ### Changed

diff --git a/covalent_awsbatch_plugin/awsbatch.py b/covalent_awsbatch_plugin/awsbatch.py
@@ -220,7 +220,10 @@ async def submit_task(self, task_metadata: Dict, identity: Dict) -> Any:
         account = identity["Account"]
 
         self._debug_log("Submitting task...")
-        batch = boto3.Session(**self.boto_session_options()).client("batch")
+        boto_session = boto3.Session(**self.boto_session_options())
+        batch = boto_session.client("batch")
+
+        region = boto_session.region_name
 
         resources = [
             {"type": "VCPU", "value": str(int(self.vcpu))},
@@ -263,7 +266,7 @@ async def submit_task(self, task_metadata: Dict, identity: Dict) -> Any:
                 "logConfiguration": {
                     "logDriver": "awslogs",
                     "options": {
-                        "awslogs-region": self.region,
+                        "awslogs-region": region,
                         "awslogs-group": self.log_group_name,
                         "awslogs-create-group": "true",
                         "awslogs-stream-prefix": "covalent-batch",