From 79d75884788484345fab54a9bbbe5c8194a9649f Mon Sep 17 00:00:00 2001 From: Parth Bansal Date: Mon, 31 Mar 2025 14:05:49 +0000 Subject: [PATCH 1/2] Fix test --- tests/test_core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_core.py b/tests/test_core.py index 9de0cb90d..a96d991b7 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -20,6 +20,7 @@ from databricks.sdk.databricks.oauth import Token from databricks.sdk.service.catalog import PermissionsChange from databricks.sdk.service.iam import AccessControlRequest +from databricks.sdk.service.jobs import JobAccessControlRequest from databricks.sdk.version import __version__ from .conftest import noop_credentials @@ -287,9 +288,8 @@ def test_access_control_list(config, requests_mock): "http://localhost/api/2.2/jobs/create", request_headers={"User-Agent": config.user_agent}, ) - w = WorkspaceClient(config=config) - res = w.jobs.create(access_control_list=[AccessControlRequest(group_name="group")]) + res = w.jobs.create(access_control_list=[JobAccessControlRequest(group_name="group")]) assert requests_mock.call_count == 1 assert requests_mock.called From 3751810dd6009532cfa78f5c39db42b66fa36560 Mon Sep 17 00:00:00 2001 From: Parth Bansal Date: Mon, 31 Mar 2025 14:08:08 +0000 Subject: [PATCH 2/2] update --- tests/test_core.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/test_core.py b/tests/test_core.py index a96d991b7..c8008df37 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -19,7 +19,6 @@ DatabricksEnvironment) from databricks.sdk.databricks.oauth import Token from databricks.sdk.service.catalog import PermissionsChange -from databricks.sdk.service.iam import AccessControlRequest from databricks.sdk.service.jobs import JobAccessControlRequest from databricks.sdk.version import __version__