add -> Connection annotation

databricks · Mar 30, 2024 · 5eedb1c · 5eedb1c
1 parent 221d8b1
commit 5eedb1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/databricks/sql/__init__.py b/src/databricks/sql/__init__.py
@@ -78,7 +78,7 @@ def TimestampFromTicks(ticks):
     return Timestamp(*time.localtime(ticks)[:6])
 
 
-def connect(server_hostname, http_path, access_token=None, **kwargs):
+def connect(server_hostname, http_path, access_token=None, **kwargs) -> "Connection":
     from .client import Connection
 
     return Connection(server_hostname, http_path, access_token, **kwargs)