Skip to content

Commit 071ad4e

Browse files
reduce changes in DatabricksRetryPolicy
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 878a845 commit 071ad4e

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

src/databricks/sql/auth/retry.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import typing
55
from importlib.metadata import version
66
from enum import Enum
7-
from typing import Any, List, Optional, Tuple, Union
7+
from typing import List, Optional, Tuple, Union
88

99
import urllib3
1010

@@ -123,15 +123,14 @@ def __init__(
123123
_total: int = urllib3_kwargs.pop("total")
124124
_attempts_remaining = _total
125125

126-
_urllib_kwargs_we_care_about: dict[str, Any] = dict(
126+
_urllib_kwargs_we_care_about = dict(
127127
total=_attempts_remaining,
128128
respect_retry_after_header=True,
129129
backoff_factor=self.delay_min,
130-
allowed_methods=["POST"],
130+
allowed_methods=["POST", "GET", "DELETE"],
131131
status_forcelist=[429, 503, *self.force_dangerous_codes],
132132
)
133-
_urllib_kwargs_we_care_about.update(**urllib3_kwargs)
134-
urllib3_kwargs = _urllib_kwargs_we_care_about
133+
urllib3_kwargs.update(**_urllib_kwargs_we_care_about)
135134

136135
super().__init__(
137136
**urllib3_kwargs,

0 commit comments

Comments
 (0)