From fb07395a115028d9611c1ea6bf608db990951ed8 Mon Sep 17 00:00:00 2001
From: Sax Authors <no-reply@google.com>
Date: Wed, 14 Aug 2024 10:49:24 -0700
Subject: [PATCH] Reduce the max retry interval from 1 min to 20 secs.

PiperOrigin-RevId: 662978128
Change-Id: I6647ef16f426b9031efe4623dee219f33cab231a
---
 saxml/common/retrier.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/saxml/common/retrier.go b/saxml/common/retrier.go
index dd526837..0c66209b 100644
--- a/saxml/common/retrier.go
+++ b/saxml/common/retrier.go
@@ -59,6 +59,8 @@ func (q queryRetrier) Do(ctx context.Context, query Closure, retriable IsRetriab
 	opts := backoff.NewExponentialBackOff()
 	// Fast initial retries.
 	opts.InitialInterval = 10 * time.Millisecond
+	// Max retry interval.
+	opts.MaxInterval = 20 * time.Second
 	// Because the retry loop still respects ctx's deadline, we want
 	// the loop retries as many times as necessary.
 	opts.MaxElapsedTime = 0