From 5de1293a219f2e17bcd3ad2bd109f48b9f68cace Mon Sep 17 00:00:00 2001
From: benpankow <ben@elementl.com>
Date: Mon, 8 Aug 2022 14:01:20 -0700
Subject: [PATCH] [deploy] Always use long agent HB timeout

---
 src/deploy.sh | 9 +--------
 1 file changed, 1 insertion(+), 8 deletions(-)

diff --git a/src/deploy.sh b/src/deploy.sh
index 9663acd7..6fc631ed 100755
--- a/src/deploy.sh
+++ b/src/deploy.sh
@@ -91,13 +91,6 @@ echo "Deploying location ${INPUT_LOCATION_NAME} to deployment ${DEPLOYMENT_NAME}
 
 echo "::set-output name=deployment::${DEPLOYMENT_NAME}"
 
-# Extend timeout in case the agent is still spinning up
-if [ $GITHUB_RUN_NUMBER -eq 1 ]; then
-    AGENT_HEARTBEAT_TIMEOUT=600
-else
-    AGENT_HEARTBEAT_TIMEOUT=90
-fi
-
 dagster-cloud workspace add-location \
     --url "${DAGSTER_CLOUD_URL}/${DEPLOYMENT_NAME}" \
     --api-token "$DAGSTER_CLOUD_API_TOKEN" \
@@ -105,7 +98,7 @@ dagster-cloud workspace add-location \
     --location-name "${INPUT_LOCATION_NAME}" \
     --image "${INPUT_REGISTRY}:${INPUT_IMAGE_TAG}" \
     --location-load-timeout 600 \
-    --agent-heartbeat-timeout $AGENT_HEARTBEAT_TIMEOUT \
+    --agent-heartbeat-timeout 600 \
     --git-url $COMMIT_URL \
     --commit-hash $GITHUB_SHA