Skip to content

Commit dab5712

Browse files
committed
rephrase import
Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>
1 parent 46075f9 commit dab5712

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/unit/test_telemetry.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -304,8 +304,8 @@ def test_connection_failure_sends_correct_telemetry_payload(
304304
mock_session.side_effect = Exception(error_message)
305305

306306
try:
307-
from databricks.sql.client import Connection
308-
Connection(server_hostname="test-host", http_path="/test-path")
307+
from databricks import sql
308+
sql.connect(server_hostname="test-host", http_path="/test-path")
309309
except Exception as e:
310310
assert str(e) == error_message
311311

0 commit comments

Comments
 (0)