MaxRetryDurationError

saishreeeee · saishreeeee · commit 341bb75aa2e0 · 2025-07-07T10:15:32.000+05:30
Signed-off-by: Sai Shree Pradhan &lt;saishree.pradhan@databricks.com&gt;
diff --git a/src/databricks/sql/client.py b/src/databricks/sql/client.py
@@ -308,7 +308,7 @@ def read(self) -> Optional[OAuthToken]:
             self._open_session_resp = self.thrift_backend.open_session(
                 session_configuration, catalog, schema
             )
-        except (RequestError, MaxRetryError) as e:
+        except (RequestError, MaxRetryError, MaxRetryDurationError) as e:
             raise
         except Exception as e:
             raise ConnectionError(

Original file line number	Diff line number	Diff line change
`@@ -308,7 +308,7 @@ def read(self) -> Optional[OAuthToken]:`
`308`	`308`	`self._open_session_resp = self.thrift_backend.open_session(`
`309`	`309`	`session_configuration, catalog, schema`
`310`	`310`	`)`
`311`		`- except (RequestError, MaxRetryError) as e:`
	`311`	`+ except (RequestError, MaxRetryError, MaxRetryDurationError) as e:`
`312`	`312`	`raise`
`313`	`313`	`except Exception as e:`
`314`	`314`	`raise ConnectionError(`