From a2b7b97219fe776c6201576b8509a500c72d0b10 Mon Sep 17 00:00:00 2001 From: Tanmay Rustagi Date: Mon, 22 Apr 2024 16:42:30 +0200 Subject: [PATCH] Update SDK to OpenAPI spec --- databricks/sdk/service/sql.py | 1 + docs/account/iam/workspace_assignment.rst | 6 +++--- docs/dbdataclasses/sql.rst | 2 ++ docs/workspace/compute/clusters.rst | 3 +++ .../update_workspace_assignment_on_aws.py | 6 +++--- 5 files changed, 12 insertions(+), 6 deletions(-) diff --git a/databricks/sdk/service/sql.py b/databricks/sdk/service/sql.py index 7bd0cc7db..c63d7c5cb 100755 --- a/databricks/sdk/service/sql.py +++ b/databricks/sdk/service/sql.py @@ -360,6 +360,7 @@ def from_dict(cls, d: Dict[str, any]) -> ChannelInfo: class ChannelName(Enum): + """Name of the channel""" CHANNEL_NAME_CURRENT = 'CHANNEL_NAME_CURRENT' CHANNEL_NAME_CUSTOM = 'CHANNEL_NAME_CUSTOM' diff --git a/docs/account/iam/workspace_assignment.rst b/docs/account/iam/workspace_assignment.rst index 1ce06996e..878ad847e 100644 --- a/docs/account/iam/workspace_assignment.rst +++ b/docs/account/iam/workspace_assignment.rst @@ -82,9 +82,9 @@ workspace_id = os.environ["DUMMY_WORKSPACE_ID"] - _ = a.workspace_assignment.update(workspace_id=workspace_id, - principal_id=spn_id, - permissions=[iam.WorkspacePermission.USER]) + a.workspace_assignment.update(workspace_id=workspace_id, + principal_id=spn_id, + permissions=[iam.WorkspacePermission.USER]) Create or update permissions assignment. diff --git a/docs/dbdataclasses/sql.rst b/docs/dbdataclasses/sql.rst index adf3ced56..fe1469a30 100644 --- a/docs/dbdataclasses/sql.rst +++ b/docs/dbdataclasses/sql.rst @@ -64,6 +64,8 @@ These dataclasses are used in the SDK to represent API requests and responses fo .. py:class:: ChannelName + Name of the channel + .. py:attribute:: CHANNEL_NAME_CURRENT :value: "CHANNEL_NAME_CURRENT" diff --git a/docs/workspace/compute/clusters.rst b/docs/workspace/compute/clusters.rst index db6f22991..437dc9cf6 100644 --- a/docs/workspace/compute/clusters.rst +++ b/docs/workspace/compute/clusters.rst @@ -845,6 +845,8 @@ :param photon_driver_capable: bool :param graviton: bool :param is_io_cache_enabled: bool + :param support_port_forwarding: bool + :param fleet: bool :returns: `node_type` compatible string @@ -868,6 +870,7 @@ :param beta: bool :param latest: bool :param ml: bool + :param genomics: bool :param gpu: bool :param scala: str :param spark_version: str diff --git a/examples/workspace_assignment/update_workspace_assignment_on_aws.py b/examples/workspace_assignment/update_workspace_assignment_on_aws.py index f12e85891..48a26a40b 100755 --- a/examples/workspace_assignment/update_workspace_assignment_on_aws.py +++ b/examples/workspace_assignment/update_workspace_assignment_on_aws.py @@ -12,6 +12,6 @@ workspace_id = os.environ["DUMMY_WORKSPACE_ID"] -_ = a.workspace_assignment.update(workspace_id=workspace_id, - principal_id=spn_id, - permissions=[iam.WorkspacePermission.USER]) +a.workspace_assignment.update(workspace_id=workspace_id, + principal_id=spn_id, + permissions=[iam.WorkspacePermission.USER])