From 53c278ea5e7d592ce2ad120b14d75951c66af086 Mon Sep 17 00:00:00 2001
From: Joey Zhao <5253430+joeyzhao2018@users.noreply.github.com>
Date: Wed, 3 Apr 2024 15:05:01 -0400
Subject: [PATCH] use sso for govcloud

---
 scripts/publish_prod.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/publish_prod.sh b/scripts/publish_prod.sh
index f88ac5e2..aa53df71 100755
--- a/scripts/publish_prod.sh
+++ b/scripts/publish_prod.sh
@@ -85,11 +85,11 @@ if [ "$CONT" != "y" ]; then
 else
     echo "Ensuring you have access to the AWS GovCloud account"
     ddsaml2aws login -a govcloud-us1-fed-human-engineering
-    AWS_PROFILE=govcloud-us1-fed-human-engineering aws sts get-caller-identity
+    aws-vault exec sso-govcloud-us1-fed-engineering -- aws sts get-caller-identity
 
     echo "Publishing layers to GovCloud AWS regions"
     ddsaml2aws login -a govcloud-us1-fed-human-engineering
-    VERSION=$LAYER_VERSION AWS_PROFILE=govcloud-us1-fed-human-engineering ./scripts/publish_layers.sh
+    VERSION=$LAYER_VERSION aws-vault exec sso-govcloud-us1-fed-engineering -- ./scripts/publish_layers.sh
 fi
 
 echo "Answer 'n' if GitLab already did this"