Merge branch 'develop' into gtc-2868

wri · Jun 26, 2024 · b0b229f · b0b229f
2 parents 7e64844 + c9bcbf5
commit b0b229f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/batch/python/export_to_gee.py b/batch/python/export_to_gee.py
@@ -1,3 +1,5 @@
+#!/usr/bin/env python
+
 import json
 import os
 
@@ -9,12 +11,13 @@
 
 EE_PROJECT = "forma-250"
 GCS_BUCKET = "data-api-gee-assets"
-GCS_SECRET_KEY_ARN = os.environ["GCS_SECRET_KEY_ARN"]
+GCS_SECRET_KEY_ARN = os.environ["AWS_GCS_KEY_SECRET_ARN"]
 GCS_CREDENTIALS_FILE = "gcs_credentials.json"
+AWS_REGION = os.environ["AWS_REGION"]
 
 
 def set_google_application_credentials():
-    client = boto3.client("secretsmanager")
+    client = boto3.client("secretsmanager", region_name=AWS_REGION)
     response = client.get_secret_value(SecretId=GCS_SECRET_KEY_ARN)
 
     with open(GCS_CREDENTIALS_FILE, "w") as f:

diff --git a/batch/scripts/cogify.sh b/batch/scripts/cogify.sh
@@ -42,7 +42,7 @@ fi
 aws s3 cp cog.tif "${TARGET}"
 
 if [ -n "$EXPORT_TO_GEE" ]; then
-  python export_to_gee.py --dataset "${DATASET}" --implementation "${IMPLEMENTATION}"
+  export_to_gee.py --dataset "${DATASET}" --implementation "${IMPLEMENTATION}"
 fi
 
 set +x