From 2a99fa0865d2a7ec47407f30a093bb28787879f7 Mon Sep 17 00:00:00 2001
From: Lin Zhou <lin.zhou@databricks.com>
Date: Wed, 9 Oct 2024 14:44:45 -0700
Subject: [PATCH] revert numRetries

---
 .../main/scala/io/delta/sharing/client/util/RetryUtils.scala    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/client/src/main/scala/io/delta/sharing/client/util/RetryUtils.scala b/client/src/main/scala/io/delta/sharing/client/util/RetryUtils.scala
index 5f176b614..bbf6fa7cb 100644
--- a/client/src/main/scala/io/delta/sharing/client/util/RetryUtils.scala
+++ b/client/src/main/scala/io/delta/sharing/client/util/RetryUtils.scala
@@ -49,7 +49,7 @@ private[sharing] object RetryUtils extends Logging {
             ", totalDuration=" + totalDuration + " : " + e.getMessage,
             e
           )
-          if (shouldRetry(e) && times <= 2 && totalDuration <= maxDurationMillis) {
+          if (shouldRetry(e) && times <= numRetries && totalDuration <= maxDurationMillis) {
             logWarning(s"Sleeping $sleepMs ms to retry on error: ${e.getMessage}.")
             sleeper(sleepMs)
             sleepMs *= 2