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

databricks/sdk/service/ml.py

@@ -716,6 +716,20 @@ class DatasetInput:
         return cls(dataset=_from_dict(d, 'dataset', Dataset), tags=_repeated_dict(d, 'tags', InputTag))
 
 
+@dataclass
+class DeleteCommentResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteCommentResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteCommentResponse:
+        """Deserializes the DeleteCommentResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class DeleteExperiment:
     experiment_id: str
@@ -733,6 +747,76 @@ class DeleteExperiment:
         return cls(experiment_id=d.get('experiment_id', None))
 
 
+@dataclass
+class DeleteExperimentResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteExperimentResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteExperimentResponse:
+        """Deserializes the DeleteExperimentResponse from a dictionary."""
+        return cls()
+
+
+@dataclass
+class DeleteModelResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteModelResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteModelResponse:
+        """Deserializes the DeleteModelResponse from a dictionary."""
+        return cls()
+
+
+@dataclass
+class DeleteModelTagResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteModelTagResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteModelTagResponse:
+        """Deserializes the DeleteModelTagResponse from a dictionary."""
+        return cls()
+
+
+@dataclass
+class DeleteModelVersionResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteModelVersionResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteModelVersionResponse:
+        """Deserializes the DeleteModelVersionResponse from a dictionary."""
+        return cls()
+
+
+@dataclass
+class DeleteModelVersionTagResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteModelVersionTagResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteModelVersionTagResponse:
+        """Deserializes the DeleteModelVersionTagResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class DeleteRun:
     run_id: str
@@ -750,6 +834,20 @@ class DeleteRun:
         return cls(run_id=d.get('run_id', None))
 
 
+@dataclass
+class DeleteRunResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteRunResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteRunResponse:
+        """Deserializes the DeleteRunResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class DeleteRuns:
     experiment_id: str
@@ -817,6 +915,34 @@ class DeleteTag:
         return cls(key=d.get('key', None), run_id=d.get('run_id', None))
 
 
+@dataclass
+class DeleteTagResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteTagResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteTagResponse:
+        """Deserializes the DeleteTagResponse from a dictionary."""
+        return cls()
+
+
+@dataclass
+class DeleteTransitionRequestResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteTransitionRequestResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteTransitionRequestResponse:
+        """Deserializes the DeleteTransitionRequestResponse from a dictionary."""
+        return cls()
+
+
 class DeleteTransitionRequestStage(Enum):
 
     ARCHIVED = 'Archived'
@@ -825,6 +951,20 @@ class DeleteTransitionRequestStage(Enum):
     STAGING = 'Staging'
 
 
+@dataclass
+class DeleteWebhookResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteWebhookResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteWebhookResponse:
+        """Deserializes the DeleteWebhookResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class Experiment:
     artifact_location: Optional[str] = None
@@ -1556,6 +1696,20 @@ class LogBatch:
                    tags=_repeated_dict(d, 'tags', RunTag))
 
 
+@dataclass
+class LogBatchResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the LogBatchResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> LogBatchResponse:
+        """Deserializes the LogBatchResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class LogInputs:
     datasets: Optional[List[DatasetInput]] = None
@@ -1577,6 +1731,20 @@ class LogInputs:
         return cls(datasets=_repeated_dict(d, 'datasets', DatasetInput), run_id=d.get('run_id', None))
 
 
+@dataclass
+class LogInputsResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the LogInputsResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> LogInputsResponse:
+        """Deserializes the LogInputsResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class LogMetric:
     key: str
@@ -1620,6 +1788,20 @@ class LogMetric:
                    value=d.get('value', None))
 
 
+@dataclass
+class LogMetricResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the LogMetricResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> LogMetricResponse:
+        """Deserializes the LogMetricResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class LogModel:
     model_json: Optional[str] = None
@@ -1641,6 +1823,20 @@ class LogModel:
         return cls(model_json=d.get('model_json', None), run_id=d.get('run_id', None))
 
 
+@dataclass
+class LogModelResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the LogModelResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> LogModelResponse:
+        """Deserializes the LogModelResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class LogParam:
     key: str
@@ -1674,6 +1870,20 @@ class LogParam:
                    value=d.get('value', None))
 
 
+@dataclass
+class LogParamResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the LogParamResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> LogParamResponse:
+        """Deserializes the LogParamResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class Metric:
     key: Optional[str] = None
@@ -2481,6 +2691,20 @@ class RestoreExperiment:
         return cls(experiment_id=d.get('experiment_id', None))
 
 
+@dataclass
+class RestoreExperimentResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the RestoreExperimentResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> RestoreExperimentResponse:
+        """Deserializes the RestoreExperimentResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class RestoreRun:
     run_id: str
@@ -2498,6 +2722,20 @@ class RestoreRun:
         return cls(run_id=d.get('run_id', None))
 
 
+@dataclass
+class RestoreRunResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the RestoreRunResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> RestoreRunResponse:
+        """Deserializes the RestoreRunResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class RestoreRuns:
     experiment_id: str
@@ -2936,6 +3174,20 @@ class SetExperimentTag:
                    value=d.get('value', None))
 
 
+@dataclass
+class SetExperimentTagResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the SetExperimentTagResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> SetExperimentTagResponse:
+        """Deserializes the SetExperimentTagResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class SetModelTagRequest:
     name: str
@@ -2964,6 +3216,20 @@ class SetModelTagRequest:
         return cls(key=d.get('key', None), name=d.get('name', None), value=d.get('value', None))
 
 
+@dataclass
+class SetModelTagResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the SetModelTagResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> SetModelTagResponse:
+        """Deserializes the SetModelTagResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class SetModelVersionTagRequest:
     name: str
@@ -2999,6 +3265,20 @@ class SetModelVersionTagRequest:
                    version=d.get('version', None))
 
 
+@dataclass
+class SetModelVersionTagResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the SetModelVersionTagResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> SetModelVersionTagResponse:
+        """Deserializes the SetModelVersionTagResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class SetTag:
     key: str
@@ -3034,6 +3314,20 @@ class SetTag:
                    value=d.get('value', None))
 
 
+@dataclass
+class SetTagResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the SetTagResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> SetTagResponse:
+        """Deserializes the SetTagResponse from a dictionary."""
+        return cls()
+
+
 class Stage(Enum):
     """Stage of the model version. Valid values are:
     
@@ -3294,6 +3588,20 @@ class UpdateExperiment:
         return cls(experiment_id=d.get('experiment_id', None), new_name=d.get('new_name', None))
 
 
+@dataclass
+class UpdateExperimentResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the UpdateExperimentResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UpdateExperimentResponse:
+        """Deserializes the UpdateExperimentResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class UpdateModelRequest:
     name: str
@@ -3315,6 +3623,20 @@ class UpdateModelRequest:
         return cls(description=d.get('description', None), name=d.get('name', None))
 
 
+@dataclass
+class UpdateModelResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the UpdateModelResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UpdateModelResponse:
+        """Deserializes the UpdateModelResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class UpdateModelVersionRequest:
     name: str
@@ -3342,6 +3664,20 @@ class UpdateModelVersionRequest:
                    version=d.get('version', None))
 
 
+@dataclass
+class UpdateModelVersionResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the UpdateModelVersionResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UpdateModelVersionResponse:
+        """Deserializes the UpdateModelVersionResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class UpdateRegistryWebhook:
     id: str
@@ -3475,6 +3811,20 @@ class UpdateRunStatus(Enum):
     SCHEDULED = 'SCHEDULED'
 
 
+@dataclass
+class UpdateWebhookResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the UpdateWebhookResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UpdateWebhookResponse:
+        """Deserializes the UpdateWebhookResponse from a dictionary."""
+        return cls()
+
+
 class ExperimentsAPI:
     """Experiments are the primary unit of organization in MLflow; all MLflow runs belong to an experiment. Each
     experiment lets you visualize, search, and compare runs, as well as download run artifacts or metadata for
@@ -3517,6 +3867,7 @@ class ExperimentsAPI:
         if name is not None: body['name'] = name
         if tags is not None: body['tags'] = [v.as_dict() for v in tags]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/experiments/create', body=body, headers=headers)
         return CreateExperimentResponse.from_dict(res)
 
@@ -3550,6 +3901,7 @@ class ExperimentsAPI:
         if tags is not None: body['tags'] = [v.as_dict() for v in tags]
         if user_id is not None: body['user_id'] = user_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/runs/create', body=body, headers=headers)
         return CreateRunResponse.from_dict(res)
 
@@ -3567,6 +3919,7 @@ class ExperimentsAPI:
         body = {}
         if experiment_id is not None: body['experiment_id'] = experiment_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/experiments/delete', body=body, headers=headers)
 
     def delete_run(self, run_id: str):
@@ -3582,6 +3935,7 @@ class ExperimentsAPI:
         body = {}
         if run_id is not None: body['run_id'] = run_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/runs/delete', body=body, headers=headers)
 
     def delete_runs(self,
@@ -3611,6 +3965,7 @@ class ExperimentsAPI:
         if max_runs is not None: body['max_runs'] = max_runs
         if max_timestamp_millis is not None: body['max_timestamp_millis'] = max_timestamp_millis
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/databricks/runs/delete-runs', body=body, headers=headers)
         return DeleteRunsResponse.from_dict(res)
 
@@ -3631,6 +3986,7 @@ class ExperimentsAPI:
         if key is not None: body['key'] = key
         if run_id is not None: body['run_id'] = run_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/runs/delete-tag', body=body, headers=headers)
 
     def get_by_name(self, experiment_name: str) -> GetExperimentResponse:
@@ -3653,6 +4009,7 @@ class ExperimentsAPI:
         query = {}
         if experiment_name is not None: query['experiment_name'] = experiment_name
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/mlflow/experiments/get-by-name', query=query, headers=headers)
         return GetExperimentResponse.from_dict(res)
 
@@ -3670,6 +4027,7 @@ class ExperimentsAPI:
         query = {}
         if experiment_id is not None: query['experiment_id'] = experiment_id
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/mlflow/experiments/get', query=query, headers=headers)
         return GetExperimentResponse.from_dict(res)
 
@@ -3729,6 +4087,7 @@ class ExperimentsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/permissions/experiments/{experiment_id}/permissionLevels',
                            headers=headers)
@@ -3746,6 +4105,7 @@ class ExperimentsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/permissions/experiments/{experiment_id}', headers=headers)
         return ExperimentPermissions.from_dict(res)
 
@@ -3770,6 +4130,7 @@ class ExperimentsAPI:
         if run_id is not None: query['run_id'] = run_id
         if run_uuid is not None: query['run_uuid'] = run_uuid
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/mlflow/runs/get', query=query, headers=headers)
         return GetRunResponse.from_dict(res)
 
@@ -3912,6 +4273,7 @@ class ExperimentsAPI:
         if run_id is not None: body['run_id'] = run_id
         if tags is not None: body['tags'] = [v.as_dict() for v in tags]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/runs/log-batch', body=body, headers=headers)
 
     def log_inputs(self, *, datasets: Optional[List[DatasetInput]] = None, run_id: Optional[str] = None):
@@ -3930,6 +4292,7 @@ class ExperimentsAPI:
         if datasets is not None: body['datasets'] = [v.as_dict() for v in datasets]
         if run_id is not None: body['run_id'] = run_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/runs/log-inputs', body=body, headers=headers)
 
     def log_metric(self,
@@ -3970,6 +4333,7 @@ class ExperimentsAPI:
         if timestamp is not None: body['timestamp'] = timestamp
         if value is not None: body['value'] = value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/runs/log-metric', body=body, headers=headers)
 
     def log_model(self, *, model_json: Optional[str] = None, run_id: Optional[str] = None):
@@ -3988,6 +4352,7 @@ class ExperimentsAPI:
         if model_json is not None: body['model_json'] = model_json
         if run_id is not None: body['run_id'] = run_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/runs/log-model', body=body, headers=headers)
 
     def log_param(self,
@@ -4020,6 +4385,7 @@ class ExperimentsAPI:
         if run_uuid is not None: body['run_uuid'] = run_uuid
         if value is not None: body['value'] = value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/runs/log-parameter', body=body, headers=headers)
 
     def restore_experiment(self, experiment_id: str):
@@ -4039,6 +4405,7 @@ class ExperimentsAPI:
         body = {}
         if experiment_id is not None: body['experiment_id'] = experiment_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/experiments/restore', body=body, headers=headers)
 
     def restore_run(self, run_id: str):
@@ -4054,6 +4421,7 @@ class ExperimentsAPI:
         body = {}
         if run_id is not None: body['run_id'] = run_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/runs/restore', body=body, headers=headers)
 
     def restore_runs(self,
@@ -4083,6 +4451,7 @@ class ExperimentsAPI:
         if max_runs is not None: body['max_runs'] = max_runs
         if min_timestamp_millis is not None: body['min_timestamp_millis'] = min_timestamp_millis
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/databricks/runs/restore-runs', body=body, headers=headers)
         return RestoreRunsResponse.from_dict(res)
 
@@ -4210,6 +4579,7 @@ class ExperimentsAPI:
         if key is not None: body['key'] = key
         if value is not None: body['value'] = value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/experiments/set-experiment-tag', body=body, headers=headers)
 
     def set_permissions(
@@ -4232,6 +4602,7 @@ class ExperimentsAPI:
         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/experiments/{experiment_id}',
                            body=body,
@@ -4263,6 +4634,7 @@ class ExperimentsAPI:
         if run_uuid is not None: body['run_uuid'] = run_uuid
         if value is not None: body['value'] = value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/runs/set-tag', body=body, headers=headers)
 
     def update_experiment(self, experiment_id: str, *, new_name: Optional[str] = None):
@@ -4281,6 +4653,7 @@ class ExperimentsAPI:
         if experiment_id is not None: body['experiment_id'] = experiment_id
         if new_name is not None: body['new_name'] = new_name
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/experiments/update', body=body, headers=headers)
 
     def update_permissions(
@@ -4303,6 +4676,7 @@ class ExperimentsAPI:
         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/experiments/{experiment_id}',
                            body=body,
@@ -4337,6 +4711,7 @@ class ExperimentsAPI:
         if run_uuid is not None: body['run_uuid'] = run_uuid
         if status is not None: body['status'] = status.value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/runs/update', body=body, headers=headers)
         return UpdateRunResponse.from_dict(res)
 
@@ -4393,6 +4768,7 @@ class ModelRegistryAPI:
         if stage is not None: body['stage'] = stage.value
         if version is not None: body['version'] = version
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/transition-requests/approve', body=body, headers=headers)
         return ApproveTransitionRequestResponse.from_dict(res)
 
@@ -4416,6 +4792,7 @@ class ModelRegistryAPI:
         if name is not None: body['name'] = name
         if version is not None: body['version'] = version
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/comments/create', body=body, headers=headers)
         return CreateCommentResponse.from_dict(res)
 
@@ -4444,6 +4821,7 @@ class ModelRegistryAPI:
         if name is not None: body['name'] = name
         if tags is not None: body['tags'] = [v.as_dict() for v in tags]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/registered-models/create', body=body, headers=headers)
         return CreateModelResponse.from_dict(res)
 
@@ -4484,6 +4862,7 @@ class ModelRegistryAPI:
         if source is not None: body['source'] = source
         if tags is not None: body['tags'] = [v.as_dict() for v in tags]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/model-versions/create', body=body, headers=headers)
         return CreateModelVersionResponse.from_dict(res)
 
@@ -4522,6 +4901,7 @@ class ModelRegistryAPI:
         if stage is not None: body['stage'] = stage.value
         if version is not None: body['version'] = version
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/transition-requests/create', body=body, headers=headers)
         return CreateTransitionRequestResponse.from_dict(res)
 
@@ -4593,6 +4973,7 @@ class ModelRegistryAPI:
         if model_name is not None: body['model_name'] = model_name
         if status is not None: body['status'] = status.value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/registry-webhooks/create', body=body, headers=headers)
         return CreateWebhookResponse.from_dict(res)
 
@@ -4609,6 +4990,7 @@ class ModelRegistryAPI:
         query = {}
         if id is not None: query['id'] = id
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', '/api/2.0/mlflow/comments/delete', query=query, headers=headers)
 
     def delete_model(self, name: str):
@@ -4625,6 +5007,7 @@ class ModelRegistryAPI:
         query = {}
         if name is not None: query['name'] = name
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', '/api/2.0/mlflow/registered-models/delete', query=query, headers=headers)
 
     def delete_model_tag(self, name: str, key: str):
@@ -4645,6 +5028,7 @@ class ModelRegistryAPI:
         if key is not None: query['key'] = key
         if name is not None: query['name'] = name
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', '/api/2.0/mlflow/registered-models/delete-tag', query=query, headers=headers)
 
     def delete_model_version(self, name: str, version: str):
@@ -4664,6 +5048,7 @@ class ModelRegistryAPI:
         if name is not None: query['name'] = name
         if version is not None: query['version'] = version
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', '/api/2.0/mlflow/model-versions/delete', query=query, headers=headers)
 
     def delete_model_version_tag(self, name: str, version: str, key: str):
@@ -4687,6 +5072,7 @@ class ModelRegistryAPI:
         if name is not None: query['name'] = name
         if version is not None: query['version'] = version
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', '/api/2.0/mlflow/model-versions/delete-tag', query=query, headers=headers)
 
     def delete_transition_request(self,
@@ -4730,6 +5116,7 @@ class ModelRegistryAPI:
         if stage is not None: query['stage'] = stage.value
         if version is not None: query['version'] = version
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', '/api/2.0/mlflow/transition-requests/delete', query=query, headers=headers)
 
     def delete_webhook(self, *, id: Optional[str] = None):
@@ -4748,6 +5135,7 @@ class ModelRegistryAPI:
         query = {}
         if id is not None: query['id'] = id
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', '/api/2.0/mlflow/registry-webhooks/delete', query=query, headers=headers)
 
     def get_latest_versions(self, name: str, *, stages: Optional[List[str]] = None) -> Iterator[ModelVersion]:
@@ -4766,6 +5154,7 @@ class ModelRegistryAPI:
         if name is not None: body['name'] = name
         if stages is not None: body['stages'] = [v for v in stages]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         json = self._api.do('POST',
                             '/api/2.0/mlflow/registered-models/get-latest-versions',
                             body=body,
@@ -4791,6 +5180,7 @@ class ModelRegistryAPI:
         query = {}
         if name is not None: query['name'] = name
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            '/api/2.0/mlflow/databricks/registered-models/get',
                            query=query,
@@ -4814,6 +5204,7 @@ class ModelRegistryAPI:
         if name is not None: query['name'] = name
         if version is not None: query['version'] = version
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/mlflow/model-versions/get', query=query, headers=headers)
         return GetModelVersionResponse.from_dict(res)
 
@@ -4834,6 +5225,7 @@ class ModelRegistryAPI:
         if name is not None: query['name'] = name
         if version is not None: query['version'] = version
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            '/api/2.0/mlflow/model-versions/get-download-uri',
                            query=query,
@@ -4852,6 +5244,7 @@ class ModelRegistryAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/permissions/registered-models/{registered_model_id}/permissionLevels',
                            headers=headers)
@@ -4870,6 +5263,7 @@ class ModelRegistryAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/permissions/registered-models/{registered_model_id}',
                            headers=headers)
@@ -4922,6 +5316,7 @@ class ModelRegistryAPI:
         if name is not None: query['name'] = name
         if version is not None: query['version'] = version
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET', '/api/2.0/mlflow/transition-requests/list', query=query, headers=headers)
         parsed = ListTransitionRequestsResponse.from_dict(json).requests
         return parsed if parsed is not None else []
@@ -4999,6 +5394,7 @@ class ModelRegistryAPI:
         if stage is not None: body['stage'] = stage.value
         if version is not None: body['version'] = version
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/transition-requests/reject', body=body, headers=headers)
         return RejectTransitionRequestResponse.from_dict(res)
 
@@ -5018,6 +5414,7 @@ class ModelRegistryAPI:
         if name is not None: body['name'] = name
         if new_name is not None: body['new_name'] = new_name
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/registered-models/rename', body=body, headers=headers)
         return RenameModelResponse.from_dict(res)
 
@@ -5129,6 +5526,7 @@ class ModelRegistryAPI:
         if name is not None: body['name'] = name
         if value is not None: body['value'] = value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/registered-models/set-tag', body=body, headers=headers)
 
     def set_model_version_tag(self, name: str, version: str, key: str, value: str):
@@ -5156,6 +5554,7 @@ class ModelRegistryAPI:
         if value is not None: body['value'] = value
         if version is not None: body['version'] = version
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('POST', '/api/2.0/mlflow/model-versions/set-tag', body=body, headers=headers)
 
     def set_permissions(
@@ -5179,6 +5578,7 @@ class ModelRegistryAPI:
         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/registered-models/{registered_model_id}',
                            body=body,
@@ -5207,6 +5607,7 @@ class ModelRegistryAPI:
         if event is not None: body['event'] = event.value
         if id is not None: body['id'] = id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/mlflow/registry-webhooks/test', body=body, headers=headers)
         return TestRegistryWebhookResponse.from_dict(res)
 
@@ -5253,6 +5654,7 @@ class ModelRegistryAPI:
         if stage is not None: body['stage'] = stage.value
         if version is not None: body['version'] = version
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST',
                            '/api/2.0/mlflow/databricks/model-versions/transition-stage',
                            body=body,
@@ -5275,6 +5677,7 @@ class ModelRegistryAPI:
         if comment is not None: body['comment'] = comment
         if id is not None: body['id'] = id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('PATCH', '/api/2.0/mlflow/comments/update', body=body, headers=headers)
         return UpdateCommentResponse.from_dict(res)
 
@@ -5294,6 +5697,7 @@ class ModelRegistryAPI:
         if description is not None: body['description'] = description
         if name is not None: body['name'] = name
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('PATCH', '/api/2.0/mlflow/registered-models/update', body=body, headers=headers)
 
     def update_model_version(self, name: str, version: str, *, description: Optional[str] = None):
@@ -5315,6 +5719,7 @@ class ModelRegistryAPI:
         if name is not None: body['name'] = name
         if version is not None: body['version'] = version
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('PATCH', '/api/2.0/mlflow/model-versions/update', body=body, headers=headers)
 
     def update_permissions(
@@ -5338,6 +5743,7 @@ class ModelRegistryAPI:
         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/registered-models/{registered_model_id}',
                            body=body,
@@ -5412,4 +5818,5 @@ class ModelRegistryAPI:
         if job_spec is not None: body['job_spec'] = job_spec.as_dict()
         if status is not None: body['status'] = status.value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('PATCH', '/api/2.0/mlflow/registry-webhooks/update', body=body, headers=headers)