databricks-sdk 0.19.1__py3-none-any.whl → 0.21.0__py3-none-any.whl

databricks/sdk/service/compute.py

@@ -64,6 +64,20 @@ class AddInstanceProfile:
                    skip_validation=d.get('skip_validation', None))
 
 
+@dataclass
+class AddResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the AddResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> AddResponse:
+        """Deserializes the AddResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class Adlsgen2Info:
     destination: str
@@ -130,7 +144,8 @@ class AwsAttributes:
     will be overridden."""
 
     ebs_volume_iops: Optional[int] = None
-    """<needs content added>"""
+    """If using gp3 volumes, what IOPS to use for the disk. If this is not set, the maximum performance
+    of a gp2 volume with the same volume size will be used."""
 
     ebs_volume_size: Optional[int] = None
     """The size of each EBS volume (in GiB) launched for each instance. For general purpose SSD, this
@@ -138,7 +153,8 @@ class AwsAttributes:
     within the range 500 - 4096."""
 
     ebs_volume_throughput: Optional[int] = None
-    """<needs content added>"""
+    """If using gp3 volumes, what throughput to use for the disk. If this is not set, the maximum
+    performance of a gp2 volume with the same volume size will be used."""
 
     ebs_volume_type: Optional[EbsVolumeType] = None
     """The type of EBS volumes that will be launched with this cluster."""
@@ -302,6 +318,20 @@ class CancelCommand:
                    context_id=d.get('contextId', None))
 
 
+@dataclass
+class CancelResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the CancelResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> CancelResponse:
+        """Deserializes the CancelResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class ChangeClusterOwner:
     cluster_id: str
@@ -323,6 +353,20 @@ class ChangeClusterOwner:
         return cls(cluster_id=d.get('cluster_id', None), owner_username=d.get('owner_username', None))
 
 
+@dataclass
+class ChangeClusterOwnerResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the ChangeClusterOwnerResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> ChangeClusterOwnerResponse:
+        """Deserializes the ChangeClusterOwnerResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class ClientsTypes:
     jobs: Optional[bool] = None
@@ -2263,6 +2307,20 @@ class DeleteCluster:
         return cls(cluster_id=d.get('cluster_id', None))
 
 
+@dataclass
+class DeleteClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteClusterResponse:
+        """Deserializes the DeleteClusterResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class DeleteInstancePool:
     instance_pool_id: str
@@ -2280,6 +2338,20 @@ class DeleteInstancePool:
         return cls(instance_pool_id=d.get('instance_pool_id', None))
 
 
+@dataclass
+class DeleteInstancePoolResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteInstancePoolResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteInstancePoolResponse:
+        """Deserializes the DeleteInstancePoolResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class DeletePolicy:
     policy_id: str
@@ -2297,6 +2369,34 @@ class DeletePolicy:
         return cls(policy_id=d.get('policy_id', None))
 
 
+@dataclass
+class DeletePolicyResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeletePolicyResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeletePolicyResponse:
+        """Deserializes the DeletePolicyResponse from a dictionary."""
+        return cls()
+
+
+@dataclass
+class DeleteResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteResponse:
+        """Deserializes the DeleteResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class DestroyContext:
     cluster_id: str
@@ -2316,6 +2416,20 @@ class DestroyContext:
         return cls(cluster_id=d.get('clusterId', None), context_id=d.get('contextId', None))
 
 
+@dataclass
+class DestroyResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DestroyResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DestroyResponse:
+        """Deserializes the DestroyResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class DiskSpec:
     disk_count: Optional[int] = None
@@ -2670,6 +2784,20 @@ class EditCluster:
                    workload_type=_from_dict(d, 'workload_type', WorkloadType))
 
 
+@dataclass
+class EditClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the EditClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> EditClusterResponse:
+        """Deserializes the EditClusterResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class EditInstancePool:
     instance_pool_id: str
@@ -2731,6 +2859,20 @@ class EditInstancePool:
                    node_type_id=d.get('node_type_id', None))
 
 
+@dataclass
+class EditInstancePoolResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the EditInstancePoolResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> EditInstancePoolResponse:
+        """Deserializes the EditInstancePoolResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class EditPolicy:
     policy_id: str
@@ -2799,6 +2941,34 @@ class EditPolicy:
                    policy_id=d.get('policy_id', None))
 
 
+@dataclass
+class EditPolicyResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the EditPolicyResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> EditPolicyResponse:
+        """Deserializes the EditPolicyResponse from a dictionary."""
+        return cls()
+
+
+@dataclass
+class EditResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the EditResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> EditResponse:
+        """Deserializes the EditResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class EventDetails:
     attributes: Optional[ClusterAttributes] = None
@@ -3686,6 +3856,20 @@ class InstallLibraries:
         return cls(cluster_id=d.get('cluster_id', None), libraries=_repeated_dict(d, 'libraries', Library))
 
 
+@dataclass
+class InstallLibrariesResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the InstallLibrariesResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> InstallLibrariesResponse:
+        """Deserializes the InstallLibrariesResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class InstancePoolAccessControlRequest:
     group_name: Optional[str] = None
@@ -4763,6 +4947,20 @@ class PermanentDeleteCluster:
         return cls(cluster_id=d.get('cluster_id', None))
 
 
+@dataclass
+class PermanentDeleteClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the PermanentDeleteClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> PermanentDeleteClusterResponse:
+        """Deserializes the PermanentDeleteClusterResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class PinCluster:
     cluster_id: str
@@ -4780,6 +4978,20 @@ class PinCluster:
         return cls(cluster_id=d.get('cluster_id', None))
 
 
+@dataclass
+class PinClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the PinClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> PinClusterResponse:
+        """Deserializes the PinClusterResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class Policy:
     created_at_timestamp: Optional[int] = None
@@ -4955,6 +5167,20 @@ class RemoveInstanceProfile:
         return cls(instance_profile_arn=d.get('instance_profile_arn', None))
 
 
+@dataclass
+class RemoveResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the RemoveResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> RemoveResponse:
+        """Deserializes the RemoveResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class ResizeCluster:
     cluster_id: str
@@ -4990,6 +5216,20 @@ class ResizeCluster:
                    num_workers=d.get('num_workers', None))
 
 
+@dataclass
+class ResizeClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the ResizeClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> ResizeClusterResponse:
+        """Deserializes the ResizeClusterResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class RestartCluster:
     cluster_id: str
@@ -5011,6 +5251,20 @@ class RestartCluster:
         return cls(cluster_id=d.get('cluster_id', None), restart_user=d.get('restart_user', None))
 
 
+@dataclass
+class RestartClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the RestartClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> RestartClusterResponse:
+        """Deserializes the RestartClusterResponse from a dictionary."""
+        return cls()
+
+
 class ResultType(Enum):
 
     ERROR = 'error'
@@ -5261,6 +5515,20 @@ class StartCluster:
         return cls(cluster_id=d.get('cluster_id', None))
 
 
+@dataclass
+class StartClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the StartClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> StartClusterResponse:
+        """Deserializes the StartClusterResponse from a dictionary."""
+        return cls()
+
+
 class State(Enum):
     """Current state of the cluster."""
 
@@ -5415,6 +5683,20 @@ class UninstallLibraries:
         return cls(cluster_id=d.get('cluster_id', None), libraries=_repeated_dict(d, 'libraries', Library))
 
 
+@dataclass
+class UninstallLibrariesResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the UninstallLibrariesResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UninstallLibrariesResponse:
+        """Deserializes the UninstallLibrariesResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class UnpinCluster:
     cluster_id: str
@@ -5432,6 +5714,34 @@ class UnpinCluster:
         return cls(cluster_id=d.get('cluster_id', None))
 
 
+@dataclass
+class UnpinClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the UnpinClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UnpinClusterResponse:
+        """Deserializes the UnpinClusterResponse from a dictionary."""
+        return cls()
+
+
+@dataclass
+class UpdateResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the UpdateResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UpdateResponse:
+        """Deserializes the UpdateResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class VolumesStorageInfo:
     destination: str
@@ -5560,6 +5870,7 @@ class ClusterPoliciesAPI:
             body['policy_family_definition_overrides'] = policy_family_definition_overrides
         if policy_family_id is not None: body['policy_family_id'] = policy_family_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/policies/clusters/create', body=body, headers=headers)
         return CreatePolicyResponse.from_dict(res)
 
@@ -5576,6 +5887,7 @@ class ClusterPoliciesAPI:
         body = {}
         if policy_id is not None: body['policy_id'] = policy_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/policies/clusters/delete', body=body, headers=headers)
 
     def edit(self,
@@ -5638,6 +5950,7 @@ class ClusterPoliciesAPI:
         if policy_family_id is not None: body['policy_family_id'] = policy_family_id
         if policy_id is not None: body['policy_id'] = policy_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/policies/clusters/edit', body=body, headers=headers)
 
     def get(self, policy_id: str) -> Policy:
@@ -5654,6 +5967,7 @@ class ClusterPoliciesAPI:
         query = {}
         if policy_id is not None: query['policy_id'] = policy_id
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/policies/clusters/get', query=query, headers=headers)
         return Policy.from_dict(res)
 
@@ -5669,6 +5983,7 @@ class ClusterPoliciesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/permissions/cluster-policies/{cluster_policy_id}/permissionLevels',
                            headers=headers)
@@ -5687,6 +6002,7 @@ class ClusterPoliciesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/permissions/cluster-policies/{cluster_policy_id}',
                            headers=headers)
@@ -5714,6 +6030,7 @@ class ClusterPoliciesAPI:
         if sort_column is not None: query['sort_column'] = sort_column.value
         if sort_order is not None: query['sort_order'] = sort_order.value
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET', '/api/2.0/policies/clusters/list', query=query, headers=headers)
         parsed = ListPoliciesResponse.from_dict(json).policies
         return parsed if parsed is not None else []
@@ -5738,6 +6055,7 @@ class ClusterPoliciesAPI:
         if access_control_list is not None:
             body['access_control_list'] = [v.as_dict() for v in access_control_list]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('PUT',
                            f'/api/2.0/permissions/cluster-policies/{cluster_policy_id}',
                            body=body,
@@ -5765,6 +6083,7 @@ class ClusterPoliciesAPI:
         if access_control_list is not None:
             body['access_control_list'] = [v.as_dict() for v in access_control_list]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('PATCH',
                            f'/api/2.0/permissions/cluster-policies/{cluster_policy_id}',
                            body=body,
@@ -5878,6 +6197,7 @@ class ClustersAPI:
         if cluster_id is not None: body['cluster_id'] = cluster_id
         if owner_username is not None: body['owner_username'] = owner_username
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/clusters/change-owner', body=body, headers=headers)
 
     def create(self,
@@ -6069,6 +6389,7 @@ class ClustersAPI:
         if ssh_public_keys is not None: body['ssh_public_keys'] = [v for v in ssh_public_keys]
         if workload_type is not None: body['workload_type'] = workload_type.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         op_response = self._api.do('POST', '/api/2.0/clusters/create', body=body, headers=headers)
         return Wait(self.wait_get_cluster_running,
                     response=CreateClusterResponse.from_dict(op_response),
@@ -6152,8 +6473,11 @@ class ClustersAPI:
         body = {}
         if cluster_id is not None: body['cluster_id'] = cluster_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
-        self._api.do('POST', '/api/2.0/clusters/delete', body=body, headers=headers)
-        return Wait(self.wait_get_cluster_terminated, cluster_id=cluster_id)
+
+        op_response = self._api.do('POST', '/api/2.0/clusters/delete', body=body, headers=headers)
+        return Wait(self.wait_get_cluster_terminated,
+                    response=DeleteClusterResponse.from_dict(op_response),
+                    cluster_id=cluster_id)
 
     def delete_and_wait(self, cluster_id: str, timeout=timedelta(minutes=20)) -> ClusterDetails:
         return self.delete(cluster_id=cluster_id).result(timeout=timeout)
@@ -6356,8 +6680,11 @@ class ClustersAPI:
         if ssh_public_keys is not None: body['ssh_public_keys'] = [v for v in ssh_public_keys]
         if workload_type is not None: body['workload_type'] = workload_type.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
-        self._api.do('POST', '/api/2.0/clusters/edit', body=body, headers=headers)
-        return Wait(self.wait_get_cluster_running, cluster_id=cluster_id)
+
+        op_response = self._api.do('POST', '/api/2.0/clusters/edit', body=body, headers=headers)
+        return Wait(self.wait_get_cluster_running,
+                    response=EditClusterResponse.from_dict(op_response),
+                    cluster_id=cluster_id)
 
     def edit_and_wait(
         self,
@@ -6490,6 +6817,7 @@ class ClustersAPI:
         query = {}
         if cluster_id is not None: query['cluster_id'] = cluster_id
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/clusters/get', query=query, headers=headers)
         return ClusterDetails.from_dict(res)
 
@@ -6505,6 +6833,7 @@ class ClustersAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/permissions/clusters/{cluster_id}/permissionLevels',
                            headers=headers)
@@ -6522,6 +6851,7 @@ class ClustersAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/permissions/clusters/{cluster_id}', headers=headers)
         return ClusterPermissions.from_dict(res)
 
@@ -6548,6 +6878,7 @@ class ClustersAPI:
         query = {}
         if can_use_client is not None: query['can_use_client'] = can_use_client
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET', '/api/2.0/clusters/list', query=query, headers=headers)
         parsed = ListClustersResponse.from_dict(json).clusters
         return parsed if parsed is not None else []
@@ -6561,6 +6892,7 @@ class ClustersAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/clusters/list-node-types', headers=headers)
         return ListNodeTypesResponse.from_dict(res)
 
@@ -6574,6 +6906,7 @@ class ClustersAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/clusters/list-zones', headers=headers)
         return ListAvailableZonesResponse.from_dict(res)
 
@@ -6594,6 +6927,7 @@ class ClustersAPI:
         body = {}
         if cluster_id is not None: body['cluster_id'] = cluster_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/clusters/permanent-delete', body=body, headers=headers)
 
     def pin(self, cluster_id: str):
@@ -6610,6 +6944,7 @@ class ClustersAPI:
         body = {}
         if cluster_id is not None: body['cluster_id'] = cluster_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/clusters/pin', body=body, headers=headers)
 
     def resize(self,
@@ -6646,8 +6981,11 @@ class ClustersAPI:
         if cluster_id is not None: body['cluster_id'] = cluster_id
         if num_workers is not None: body['num_workers'] = num_workers
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
-        self._api.do('POST', '/api/2.0/clusters/resize', body=body, headers=headers)
-        return Wait(self.wait_get_cluster_running, cluster_id=cluster_id)
+
+        op_response = self._api.do('POST', '/api/2.0/clusters/resize', body=body, headers=headers)
+        return Wait(self.wait_get_cluster_running,
+                    response=ResizeClusterResponse.from_dict(op_response),
+                    cluster_id=cluster_id)
 
     def resize_and_wait(self,
                         cluster_id: str,
@@ -6677,8 +7015,11 @@ class ClustersAPI:
         if cluster_id is not None: body['cluster_id'] = cluster_id
         if restart_user is not None: body['restart_user'] = restart_user
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
-        self._api.do('POST', '/api/2.0/clusters/restart', body=body, headers=headers)
-        return Wait(self.wait_get_cluster_running, cluster_id=cluster_id)
+
+        op_response = self._api.do('POST', '/api/2.0/clusters/restart', body=body, headers=headers)
+        return Wait(self.wait_get_cluster_running,
+                    response=RestartClusterResponse.from_dict(op_response),
+                    cluster_id=cluster_id)
 
     def restart_and_wait(self,
                          cluster_id: str,
@@ -6706,6 +7047,7 @@ class ClustersAPI:
         if access_control_list is not None:
             body['access_control_list'] = [v.as_dict() for v in access_control_list]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('PUT', f'/api/2.0/permissions/clusters/{cluster_id}', body=body, headers=headers)
         return ClusterPermissions.from_dict(res)
 
@@ -6718,6 +7060,7 @@ class ClustersAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/clusters/spark-versions', headers=headers)
         return GetSparkVersionsResponse.from_dict(res)
 
@@ -6741,8 +7084,11 @@ class ClustersAPI:
         body = {}
         if cluster_id is not None: body['cluster_id'] = cluster_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
-        self._api.do('POST', '/api/2.0/clusters/start', body=body, headers=headers)
-        return Wait(self.wait_get_cluster_running, cluster_id=cluster_id)
+
+        op_response = self._api.do('POST', '/api/2.0/clusters/start', body=body, headers=headers)
+        return Wait(self.wait_get_cluster_running,
+                    response=StartClusterResponse.from_dict(op_response),
+                    cluster_id=cluster_id)
 
     def start_and_wait(self, cluster_id: str, timeout=timedelta(minutes=20)) -> ClusterDetails:
         return self.start(cluster_id=cluster_id).result(timeout=timeout)
@@ -6762,6 +7108,7 @@ class ClustersAPI:
         body = {}
         if cluster_id is not None: body['cluster_id'] = cluster_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/clusters/unpin', body=body, headers=headers)
 
     def update_permissions(
@@ -6783,6 +7130,7 @@ class ClustersAPI:
         if access_control_list is not None:
             body['access_control_list'] = [v.as_dict() for v in access_control_list]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('PATCH', f'/api/2.0/permissions/clusters/{cluster_id}', body=body, headers=headers)
         return ClusterPermissions.from_dict(res)
 
@@ -6917,8 +7265,10 @@ class CommandExecutionAPI:
         if command_id is not None: body['commandId'] = command_id
         if context_id is not None: body['contextId'] = context_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
-        self._api.do('POST', '/api/1.2/commands/cancel', body=body, headers=headers)
+
+        op_response = self._api.do('POST', '/api/1.2/commands/cancel', body=body, headers=headers)
         return Wait(self.wait_command_status_command_execution_cancelled,
+                    response=CancelResponse.from_dict(op_response),
                     cluster_id=cluster_id,
                     command_id=command_id,
                     context_id=context_id)
@@ -6952,6 +7302,7 @@ class CommandExecutionAPI:
         if command_id is not None: query['commandId'] = command_id
         if context_id is not None: query['contextId'] = context_id
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/1.2/commands/status', query=query, headers=headers)
         return CommandStatusResponse.from_dict(res)
 
@@ -6970,6 +7321,7 @@ class CommandExecutionAPI:
         if cluster_id is not None: query['clusterId'] = cluster_id
         if context_id is not None: query['contextId'] = context_id
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/1.2/contexts/status', query=query, headers=headers)
         return ContextStatusResponse.from_dict(res)
 
@@ -6995,6 +7347,7 @@ class CommandExecutionAPI:
         if cluster_id is not None: body['clusterId'] = cluster_id
         if language is not None: body['language'] = language.value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         op_response = self._api.do('POST', '/api/1.2/contexts/create', body=body, headers=headers)
         return Wait(self.wait_context_status_command_execution_running,
                     response=Created.from_dict(op_response),
@@ -7023,6 +7376,7 @@ class CommandExecutionAPI:
         if cluster_id is not None: body['clusterId'] = cluster_id
         if context_id is not None: body['contextId'] = context_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/1.2/contexts/destroy', body=body, headers=headers)
 
     def execute(self,
@@ -7055,6 +7409,7 @@ class CommandExecutionAPI:
         if context_id is not None: body['contextId'] = context_id
         if language is not None: body['language'] = language.value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         op_response = self._api.do('POST', '/api/1.2/commands/execute', body=body, headers=headers)
         return Wait(self.wait_command_status_command_execution_finished_or_error,
                     response=Created.from_dict(op_response),
@@ -7121,6 +7476,7 @@ class GlobalInitScriptsAPI:
         if position is not None: body['position'] = position
         if script is not None: body['script'] = script
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/global-init-scripts', body=body, headers=headers)
         return CreateResponse.from_dict(res)
 
@@ -7136,6 +7492,7 @@ class GlobalInitScriptsAPI:
         """
 
         headers = {}
+
         self._api.do('DELETE', f'/api/2.0/global-init-scripts/{script_id}', headers=headers)
 
     def get(self, script_id: str) -> GlobalInitScriptDetailsWithContent:
@@ -7150,6 +7507,7 @@ class GlobalInitScriptsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/global-init-scripts/{script_id}', headers=headers)
         return GlobalInitScriptDetailsWithContent.from_dict(res)
 
@@ -7164,6 +7522,7 @@ class GlobalInitScriptsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET', '/api/2.0/global-init-scripts', headers=headers)
         parsed = ListGlobalInitScriptsResponse.from_dict(json).scripts
         return parsed if parsed is not None else []
@@ -7207,6 +7566,7 @@ class GlobalInitScriptsAPI:
         if position is not None: body['position'] = position
         if script is not None: body['script'] = script
         headers = {'Content-Type': 'application/json', }
+
         self._api.do('PATCH', f'/api/2.0/global-init-scripts/{script_id}', body=body, headers=headers)
 
 
@@ -7315,6 +7675,7 @@ class InstancePoolsAPI:
         if preloaded_spark_versions is not None:
             body['preloaded_spark_versions'] = [v for v in preloaded_spark_versions]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/instance-pools/create', body=body, headers=headers)
         return CreateInstancePoolResponse.from_dict(res)
 
@@ -7331,6 +7692,7 @@ class InstancePoolsAPI:
         body = {}
         if instance_pool_id is not None: body['instance_pool_id'] = instance_pool_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/instance-pools/delete', body=body, headers=headers)
 
     def edit(self,
@@ -7386,6 +7748,7 @@ class InstancePoolsAPI:
         if min_idle_instances is not None: body['min_idle_instances'] = min_idle_instances
         if node_type_id is not None: body['node_type_id'] = node_type_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/instance-pools/edit', body=body, headers=headers)
 
     def get(self, instance_pool_id: str) -> GetInstancePool:
@@ -7402,6 +7765,7 @@ class InstancePoolsAPI:
         query = {}
         if instance_pool_id is not None: query['instance_pool_id'] = instance_pool_id
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/instance-pools/get', query=query, headers=headers)
         return GetInstancePool.from_dict(res)
 
@@ -7417,6 +7781,7 @@ class InstancePoolsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/permissions/instance-pools/{instance_pool_id}/permissionLevels',
                            headers=headers)
@@ -7435,6 +7800,7 @@ class InstancePoolsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/permissions/instance-pools/{instance_pool_id}', headers=headers)
         return InstancePoolPermissions.from_dict(res)
 
@@ -7447,6 +7813,7 @@ class InstancePoolsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET', '/api/2.0/instance-pools/list', headers=headers)
         parsed = ListInstancePools.from_dict(json).instance_pools
         return parsed if parsed is not None else []
@@ -7471,6 +7838,7 @@ class InstancePoolsAPI:
         if access_control_list is not None:
             body['access_control_list'] = [v.as_dict() for v in access_control_list]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('PUT',
                            f'/api/2.0/permissions/instance-pools/{instance_pool_id}',
                            body=body,
@@ -7498,6 +7866,7 @@ class InstancePoolsAPI:
         if access_control_list is not None:
             body['access_control_list'] = [v.as_dict() for v in access_control_list]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('PATCH',
                            f'/api/2.0/permissions/instance-pools/{instance_pool_id}',
                            body=body,
@@ -7556,6 +7925,7 @@ class InstanceProfilesAPI:
         if is_meta_instance_profile is not None: body['is_meta_instance_profile'] = is_meta_instance_profile
         if skip_validation is not None: body['skip_validation'] = skip_validation
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/instance-profiles/add', body=body, headers=headers)
 
     def edit(self,
@@ -7601,6 +7971,7 @@ class InstanceProfilesAPI:
         if instance_profile_arn is not None: body['instance_profile_arn'] = instance_profile_arn
         if is_meta_instance_profile is not None: body['is_meta_instance_profile'] = is_meta_instance_profile
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/instance-profiles/edit', body=body, headers=headers)
 
     def list(self) -> Iterator[InstanceProfile]:
@@ -7614,6 +7985,7 @@ class InstanceProfilesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET', '/api/2.0/instance-profiles/list', headers=headers)
         parsed = ListInstanceProfilesResponse.from_dict(json).instance_profiles
         return parsed if parsed is not None else []
@@ -7634,6 +8006,7 @@ class InstanceProfilesAPI:
         body = {}
         if instance_profile_arn is not None: body['instance_profile_arn'] = instance_profile_arn
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/instance-profiles/remove', body=body, headers=headers)
 
 
@@ -7669,6 +8042,7 @@ class LibrariesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/libraries/all-cluster-statuses', headers=headers)
         return ListAllClusterLibraryStatusesResponse.from_dict(res)
 
@@ -7697,6 +8071,7 @@ class LibrariesAPI:
         query = {}
         if cluster_id is not None: query['cluster_id'] = cluster_id
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET', '/api/2.0/libraries/cluster-status', query=query, headers=headers)
         parsed = ClusterLibraryStatuses.from_dict(json).library_statuses
         return parsed if parsed is not None else []
@@ -7721,6 +8096,7 @@ class LibrariesAPI:
         if cluster_id is not None: body['cluster_id'] = cluster_id
         if libraries is not None: body['libraries'] = [v.as_dict() for v in libraries]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/libraries/install', body=body, headers=headers)
 
     def uninstall(self, cluster_id: str, libraries: List[Library]):
@@ -7741,6 +8117,7 @@ class LibrariesAPI:
         if cluster_id is not None: body['cluster_id'] = cluster_id
         if libraries is not None: body['libraries'] = [v.as_dict() for v in libraries]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/libraries/uninstall', body=body, headers=headers)
 
 
@@ -7769,6 +8146,7 @@ class PolicyFamiliesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/policy-families/{policy_family_id}', headers=headers)
         return PolicyFamily.from_dict(res)