From 4444997f53aeb0a4e06e7a5d6f5ece241631ec21 Mon Sep 17 00:00:00 2001 From: Alexsander Hamir Date: Thu, 1 Jan 2026 10:33:57 -0800 Subject: [PATCH] fix: correct SDK patch paths in databricks tests - patch databricks.sdk directly instead of module attributes --- .../test_databricks_partner_integration.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/tests/test_litellm/llms/databricks/test_databricks_partner_integration.py b/tests/test_litellm/llms/databricks/test_databricks_partner_integration.py index 721e463607f..9bcd62c14b0 100644 --- a/tests/test_litellm/llms/databricks/test_databricks_partner_integration.py +++ b/tests/test_litellm/llms/databricks/test_databricks_partner_integration.py @@ -348,11 +348,9 @@ class TestSDKPartnerTelemetry: "Authorization": "Bearer token" } - with patch( - "litellm.llms.databricks.common_utils.WorkspaceClient", return_value=mock_workspace_client - ): - mock_useragent = MagicMock() - with patch("litellm.llms.databricks.common_utils.useragent", mock_useragent): + mock_useragent = MagicMock() + with patch("databricks.sdk.WorkspaceClient", return_value=mock_workspace_client): + with patch("databricks.sdk.useragent", mock_useragent): databricks_base._get_databricks_credentials( api_key=None, api_base=None, @@ -594,10 +592,8 @@ class TestAuthenticationPriority: "Authorization": "Bearer sdk-token" } - with patch( - "litellm.llms.databricks.common_utils.WorkspaceClient", return_value=mock_workspace_client - ): - with patch("litellm.llms.databricks.common_utils.useragent"): + with patch("databricks.sdk.WorkspaceClient", return_value=mock_workspace_client): + with patch("databricks.sdk.useragent"): api_base, headers = databricks_base.databricks_validate_environment( api_key=None, api_base=None,