fix tests

parthban-db · parthban-db · commit 7279a2fee9a9 · 2025-03-26T21:27:32.000Z
diff --git a/tests/test_model_serving_auth.py b/tests/test_model_serving_auth.py
@@ -124,7 +124,7 @@ def test_model_serving_auth_refresh(monkeypatch, mocker):
         "databricks.sdk.databricks.credentials_provider.ModelServingAuthProvider._MODEL_DEPENDENCY_OAUTH_TOKEN_FILE_PATH",
         "tests/testdata/model-serving-test-token",
     )
-    mocker.patch("databricks.databricks.sdk.config.Config._known_file_config_loader")
+    mocker.patch("databricks.sdk.databricks.config.Config._known_file_config_loader")
 
     cfg = Config()
     assert cfg.auth_type == "model-serving"

Original file line number	Diff line number	Diff line change
`@@ -124,7 +124,7 @@ def test_model_serving_auth_refresh(monkeypatch, mocker):`
`124`	`124`	`"databricks.sdk.databricks.credentials_provider.ModelServingAuthProvider._MODEL_DEPENDENCY_OAUTH_TOKEN_FILE_PATH",`
`125`	`125`	`"tests/testdata/model-serving-test-token",`
`126`	`126`	`)`
`127`		`- mocker.patch("databricks.databricks.sdk.config.Config._known_file_config_loader")`
	`127`	`+ mocker.patch("databricks.sdk.databricks.config.Config._known_file_config_loader")`
`128`	`128`
`129`	`129`	`cfg = Config()`
`130`	`130`	`assert cfg.auth_type == "model-serving"`