From a6749f91ee3a408b934a29a45a762ad88fc8e8e7 Mon Sep 17 00:00:00 2001
From: Taleb Zeghmi <talebz@zillowgroup.com>
Date: Wed, 10 Jan 2024 15:29:03 -0800
Subject: [PATCH] BACKOFF_DURATION_INT = 2

---
 metaflow/plugins/aip/aip_constants.py          | 2 +-
 metaflow/plugins/aip/exit_handler_decorator.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/metaflow/plugins/aip/aip_constants.py b/metaflow/plugins/aip/aip_constants.py
index be0f3c4fd1..cd1d4a8ac1 100644
--- a/metaflow/plugins/aip/aip_constants.py
+++ b/metaflow/plugins/aip/aip_constants.py
@@ -14,7 +14,7 @@
 S3_SENSOR_RETRY_COUNT = 7
 PVC_CREATE_RETRY_COUNT = 7
 EXIT_HANDLER_RETRY_COUNT = 7
-BACKOFF_DURATION_INT = "2"
+BACKOFF_DURATION_INT = 2
 BACKOFF_DURATION = f"{BACKOFF_DURATION_INT}m"
 RETRY_BACKOFF_FACTOR = 3  # 2 * 3 * 3 * 3 = 54 minutes
 
diff --git a/metaflow/plugins/aip/exit_handler_decorator.py b/metaflow/plugins/aip/exit_handler_decorator.py
index 52fbfcfe1a..243f7e6353 100644
--- a/metaflow/plugins/aip/exit_handler_decorator.py
+++ b/metaflow/plugins/aip/exit_handler_decorator.py
@@ -39,7 +39,7 @@ def wrapped(*args, **kwargs):
 
 def exit_handler_retry(
     times: int = EXIT_HANDLER_RETRY_COUNT,
-    minutes_between_retries: int = int(BACKOFF_DURATION_INT),
+    minutes_between_retries: int = BACKOFF_DURATION_INT,
     retry_backoff_factor: int = RETRY_BACKOFF_FACTOR,
 ):
     """