ai: apply changes for #876 (1 review thread)

Addresses:
  - #3636633446 at src/databricks/sql/auth/oauth.py:85

Signed-off-by: peco-engineer-bot[bot] <peco-engineer-bot[bot]@users.noreply.github.com>

Author: peco-engineer-bot[bot]

src/databricks/sql/auth/oauth.py

@@ -73,7 +73,9 @@ class OAuthManager:
     # ``redirect_callback_timeout_seconds`` argument below is an internal-only
     # override for callers constructing ``OAuthManager`` directly (private API);
     # it is not plumbed through ``DatabricksOAuthProvider`` or the public
-    # connection kwargs.
+    # connection kwargs. It accepts a ``float`` so callers (and tests) may pass
+    # fractional-second timeouts; this class default is a whole number of
+    # seconds.
     REDIRECT_CALLBACK_TIMEOUT_SECONDS = 60 * 5
 
     def __init__(
@@ -82,7 +84,7 @@ def __init__(
         client_id: str,
         idp_endpoint: OAuthEndpointCollection,
         http_client,
-        redirect_callback_timeout_seconds: Optional[int] = None,
+        redirect_callback_timeout_seconds: Optional[float] = None,
     ):
         self.port_range = port_range
         self.client_id = client_id