databricks-sdk 0.70.0__py3-none-any.whl → 0.72.0__py3-none-any.whl

databricks/sdk/service/ml.py

@@ -201,6 +201,31 @@ class ApproveTransitionRequestResponse:
         return cls(activity=_from_dict(d, "activity", Activity))
 
 
+@dataclass
+class BatchCreateMaterializedFeaturesResponse:
+    materialized_features: Optional[List[MaterializedFeature]] = None
+    """The created materialized features with assigned IDs."""
+
+    def as_dict(self) -> dict:
+        """Serializes the BatchCreateMaterializedFeaturesResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.materialized_features:
+            body["materialized_features"] = [v.as_dict() for v in self.materialized_features]
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the BatchCreateMaterializedFeaturesResponse into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.materialized_features:
+            body["materialized_features"] = self.materialized_features
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> BatchCreateMaterializedFeaturesResponse:
+        """Deserializes the BatchCreateMaterializedFeaturesResponse from a dictionary."""
+        return cls(materialized_features=_repeated_dict(d, "materialized_features", MaterializedFeature))
+
+
 class CommentActivityAction(Enum):
     """An action that a user (with sufficient permissions) could take on an activity or comment.
 
@@ -291,6 +316,38 @@ class CommentObject:
         )
 
 
+@dataclass
+class ContinuousWindow:
+    window_duration: str
+    """The duration of the continuous window (must be positive)."""
+
+    offset: Optional[str] = None
+    """The offset of the continuous window (must be non-positive)."""
+
+    def as_dict(self) -> dict:
+        """Serializes the ContinuousWindow into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.offset is not None:
+            body["offset"] = self.offset
+        if self.window_duration is not None:
+            body["window_duration"] = self.window_duration
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the ContinuousWindow into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.offset is not None:
+            body["offset"] = self.offset
+        if self.window_duration is not None:
+            body["window_duration"] = self.window_duration
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> ContinuousWindow:
+        """Deserializes the ContinuousWindow from a dictionary."""
+        return cls(offset=d.get("offset", None), window_duration=d.get("window_duration", None))
+
+
 @dataclass
 class CreateCommentResponse:
     comment: Optional[CommentObject] = None
@@ -391,6 +448,31 @@ class CreateLoggedModelResponse:
         return cls(model=_from_dict(d, "model", LoggedModel))
 
 
+@dataclass
+class CreateMaterializedFeatureRequest:
+    materialized_feature: MaterializedFeature
+    """The materialized feature to create."""
+
+    def as_dict(self) -> dict:
+        """Serializes the CreateMaterializedFeatureRequest into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.materialized_feature:
+            body["materialized_feature"] = self.materialized_feature.as_dict()
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the CreateMaterializedFeatureRequest into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.materialized_feature:
+            body["materialized_feature"] = self.materialized_feature
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> CreateMaterializedFeatureRequest:
+        """Deserializes the CreateMaterializedFeatureRequest from a dictionary."""
+        return cls(materialized_feature=_from_dict(d, "materialized_feature", MaterializedFeature))
+
+
 @dataclass
 class CreateModelResponse:
     registered_model: Optional[Model] = None
@@ -1302,6 +1384,9 @@ class Feature:
     filter_condition: Optional[str] = None
     """The filter condition applied to the source data before aggregation."""
 
+    lineage_context: Optional[LineageContext] = None
+    """Lineage context information for this feature."""
+
     def as_dict(self) -> dict:
         """Serializes the Feature into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -1315,6 +1400,8 @@ class Feature:
             body["function"] = self.function.as_dict()
         if self.inputs:
             body["inputs"] = [v for v in self.inputs]
+        if self.lineage_context:
+            body["lineage_context"] = self.lineage_context.as_dict()
         if self.source:
             body["source"] = self.source.as_dict()
         if self.time_window:
@@ -1334,6 +1421,8 @@ class Feature:
             body["function"] = self.function
         if self.inputs:
             body["inputs"] = self.inputs
+        if self.lineage_context:
+            body["lineage_context"] = self.lineage_context
         if self.source:
             body["source"] = self.source
         if self.time_window:
@@ -1349,6 +1438,7 @@ class Feature:
             full_name=d.get("full_name", None),
             function=_from_dict(d, "function", Function),
             inputs=d.get("inputs", None),
+            lineage_context=_from_dict(d, "lineage_context", LineageContext),
             source=_from_dict(d, "source", DataSource),
             time_window=_from_dict(d, "time_window", TimeWindow),
         )
@@ -2160,6 +2250,38 @@ class InputTag:
         return cls(key=d.get("key", None), value=d.get("value", None))
 
 
+@dataclass
+class JobContext:
+    job_id: Optional[int] = None
+    """The job ID where this API invoked."""
+
+    job_run_id: Optional[int] = None
+    """The job run ID where this API was invoked."""
+
+    def as_dict(self) -> dict:
+        """Serializes the JobContext into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.job_id is not None:
+            body["job_id"] = self.job_id
+        if self.job_run_id is not None:
+            body["job_run_id"] = self.job_run_id
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the JobContext into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.job_id is not None:
+            body["job_id"] = self.job_id
+        if self.job_run_id is not None:
+            body["job_run_id"] = self.job_run_id
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> JobContext:
+        """Deserializes the JobContext from a dictionary."""
+        return cls(job_id=d.get("job_id", None), job_run_id=d.get("job_run_id", None))
+
+
 @dataclass
 class JobSpec:
     job_id: str
@@ -2237,6 +2359,42 @@ class JobSpecWithoutSecret:
         return cls(job_id=d.get("job_id", None), workspace_url=d.get("workspace_url", None))
 
 
+@dataclass
+class LineageContext:
+    """Lineage context information for tracking where an API was invoked. This will allow us to track
+    lineage, which currently uses caller entity information for use across the Lineage Client and
+    Observability in Lumberjack."""
+
+    job_context: Optional[JobContext] = None
+    """Job context information including job ID and run ID."""
+
+    notebook_id: Optional[int] = None
+    """The notebook ID where this API was invoked."""
+
+    def as_dict(self) -> dict:
+        """Serializes the LineageContext into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.job_context:
+            body["job_context"] = self.job_context.as_dict()
+        if self.notebook_id is not None:
+            body["notebook_id"] = self.notebook_id
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the LineageContext into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.job_context:
+            body["job_context"] = self.job_context
+        if self.notebook_id is not None:
+            body["notebook_id"] = self.notebook_id
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> LineageContext:
+        """Deserializes the LineageContext from a dictionary."""
+        return cls(job_context=_from_dict(d, "job_context", JobContext), notebook_id=d.get("notebook_id", None))
+
+
 @dataclass
 class LinkedFeature:
     """Feature for model version. ([ML-57150] Renamed from Feature to LinkedFeature)"""
@@ -2987,10 +3145,6 @@ class MaterializedFeature:
     feature_name: str
     """The full name of the feature in Unity Catalog."""
 
-    offline_store_config: OfflineStoreConfig
-
-    online_store_config: OnlineStore
-
     last_materialization_time: Optional[str] = None
     """The timestamp when the pipeline last ran and updated the materialized feature values. If the
     pipeline has not run yet, this field will be null."""
@@ -2998,6 +3152,10 @@ class MaterializedFeature:
     materialized_feature_id: Optional[str] = None
     """Unique identifier for the materialized feature."""
 
+    offline_store_config: Optional[OfflineStoreConfig] = None
+
+    online_store_config: Optional[OnlineStore] = None
+
     pipeline_schedule_state: Optional[MaterializedFeaturePipelineScheduleState] = None
     """The schedule state of the materialization pipeline."""
 
@@ -5114,6 +5272,38 @@ class SetTagResponse:
         return cls()
 
 
+@dataclass
+class SlidingWindow:
+    window_duration: str
+    """The duration of the sliding window."""
+
+    slide_duration: str
+    """The slide duration (interval by which windows advance, must be positive and less than duration)."""
+
+    def as_dict(self) -> dict:
+        """Serializes the SlidingWindow into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.slide_duration is not None:
+            body["slide_duration"] = self.slide_duration
+        if self.window_duration is not None:
+            body["window_duration"] = self.window_duration
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the SlidingWindow into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.slide_duration is not None:
+            body["slide_duration"] = self.slide_duration
+        if self.window_duration is not None:
+            body["window_duration"] = self.window_duration
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> SlidingWindow:
+        """Deserializes the SlidingWindow from a dictionary."""
+        return cls(slide_duration=d.get("slide_duration", None), window_duration=d.get("window_duration", None))
+
+
 class Status(Enum):
     """The status of the model version. Valid values are: * `PENDING_REGISTRATION`: Request to register
     a new model version is pending as server performs background tasks.
@@ -5161,34 +5351,42 @@ class TestRegistryWebhookResponse:
 
 @dataclass
 class TimeWindow:
-    duration: str
-    """The duration of the time window."""
+    continuous: Optional[ContinuousWindow] = None
 
-    offset: Optional[str] = None
-    """The offset of the time window."""
+    sliding: Optional[SlidingWindow] = None
+
+    tumbling: Optional[TumblingWindow] = None
 
     def as_dict(self) -> dict:
         """Serializes the TimeWindow into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.duration is not None:
-            body["duration"] = self.duration
-        if self.offset is not None:
-            body["offset"] = self.offset
+        if self.continuous:
+            body["continuous"] = self.continuous.as_dict()
+        if self.sliding:
+            body["sliding"] = self.sliding.as_dict()
+        if self.tumbling:
+            body["tumbling"] = self.tumbling.as_dict()
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the TimeWindow into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.duration is not None:
-            body["duration"] = self.duration
-        if self.offset is not None:
-            body["offset"] = self.offset
+        if self.continuous:
+            body["continuous"] = self.continuous
+        if self.sliding:
+            body["sliding"] = self.sliding
+        if self.tumbling:
+            body["tumbling"] = self.tumbling
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> TimeWindow:
         """Deserializes the TimeWindow from a dictionary."""
-        return cls(duration=d.get("duration", None), offset=d.get("offset", None))
+        return cls(
+            continuous=_from_dict(d, "continuous", ContinuousWindow),
+            sliding=_from_dict(d, "sliding", SlidingWindow),
+            tumbling=_from_dict(d, "tumbling", TumblingWindow),
+        )
 
 
 @dataclass
@@ -5286,6 +5484,31 @@ class TransitionStageResponse:
         return cls(model_version_databricks=_from_dict(d, "model_version_databricks", ModelVersionDatabricks))
 
 
+@dataclass
+class TumblingWindow:
+    window_duration: str
+    """The duration of each tumbling window (non-overlapping, fixed-duration windows)."""
+
+    def as_dict(self) -> dict:
+        """Serializes the TumblingWindow into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.window_duration is not None:
+            body["window_duration"] = self.window_duration
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the TumblingWindow into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.window_duration is not None:
+            body["window_duration"] = self.window_duration
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> TumblingWindow:
+        """Deserializes the TumblingWindow from a dictionary."""
+        return cls(window_duration=d.get("window_duration", None))
+
+
 @dataclass
 class UpdateCommentResponse:
     comment: Optional[CommentObject] = None
@@ -5478,6 +5701,7 @@ class ExperimentsAPI:
 
         :returns: :class:`CreateExperimentResponse`
         """
+
         body = {}
         if artifact_location is not None:
             body["artifact_location"] = artifact_location
@@ -5520,6 +5744,7 @@ class ExperimentsAPI:
 
         :returns: :class:`CreateLoggedModelResponse`
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -5568,6 +5793,7 @@ class ExperimentsAPI:
 
         :returns: :class:`CreateRunResponse`
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -5596,6 +5822,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -5646,6 +5873,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if run_id is not None:
             body["run_id"] = run_id
@@ -5674,6 +5902,7 @@ class ExperimentsAPI:
 
         :returns: :class:`DeleteRunsResponse`
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -5700,6 +5929,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if key is not None:
             body["key"] = key
@@ -5723,6 +5953,7 @@ class ExperimentsAPI:
 
         :returns: :class:`FinalizeLoggedModelResponse`
         """
+
         body = {}
         if status is not None:
             body["status"] = status.value
@@ -6061,6 +6292,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if metrics is not None:
             body["metrics"] = [v.as_dict() for v in metrics]
@@ -6091,6 +6323,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if datasets is not None:
             body["datasets"] = [v.as_dict() for v in datasets]
@@ -6117,6 +6350,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if params is not None:
             body["params"] = [v.as_dict() for v in params]
@@ -6168,6 +6402,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if dataset_digest is not None:
             body["dataset_digest"] = dataset_digest
@@ -6207,6 +6442,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if model_json is not None:
             body["model_json"] = model_json
@@ -6229,6 +6465,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if models is not None:
             body["models"] = [v.as_dict() for v in models]
@@ -6258,6 +6495,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if key is not None:
             body["key"] = key
@@ -6286,6 +6524,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -6306,6 +6545,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if run_id is not None:
             body["run_id"] = run_id
@@ -6334,6 +6574,7 @@ class ExperimentsAPI:
 
         :returns: :class:`RestoreRunsResponse`
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -6375,6 +6616,7 @@ class ExperimentsAPI:
 
         :returns: Iterator over :class:`Experiment`
         """
+
         body = {}
         if filter is not None:
             body["filter"] = filter
@@ -6434,6 +6676,7 @@ class ExperimentsAPI:
 
         :returns: :class:`SearchLoggedModelsResponse`
         """
+
         body = {}
         if datasets is not None:
             body["datasets"] = [v.as_dict() for v in datasets]
@@ -6497,6 +6740,7 @@ class ExperimentsAPI:
 
         :returns: Iterator over :class:`Run`
         """
+
         body = {}
         if experiment_ids is not None:
             body["experiment_ids"] = [v for v in experiment_ids]
@@ -6536,6 +6780,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -6560,6 +6805,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if tags is not None:
             body["tags"] = [v.as_dict() for v in tags]
@@ -6582,6 +6828,7 @@ class ExperimentsAPI:
 
         :returns: :class:`ExperimentPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -6608,6 +6855,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if key is not None:
             body["key"] = key
@@ -6634,6 +6882,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -6657,6 +6906,7 @@ class ExperimentsAPI:
 
         :returns: :class:`ExperimentPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -6693,6 +6943,7 @@ class ExperimentsAPI:
 
         :returns: :class:`UpdateRunResponse`
         """
+
         body = {}
         if end_time is not None:
             body["end_time"] = end_time
@@ -6719,6 +6970,30 @@ class FeatureEngineeringAPI:
     def __init__(self, api_client):
         self._api = api_client
 
+    def batch_create_materialized_features(
+        self, requests: List[CreateMaterializedFeatureRequest]
+    ) -> BatchCreateMaterializedFeaturesResponse:
+        """Batch create materialized features.
+
+        :param requests: List[:class:`CreateMaterializedFeatureRequest`]
+          The requests to create materialized features.
+
+        :returns: :class:`BatchCreateMaterializedFeaturesResponse`
+        """
+
+        body = {}
+        if requests is not None:
+            body["requests"] = [v.as_dict() for v in requests]
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
+
+        res = self._api.do(
+            "POST", "/api/2.0/feature-engineering/materialized-features:batchCreate", body=body, headers=headers
+        )
+        return BatchCreateMaterializedFeaturesResponse.from_dict(res)
+
     def create_feature(self, feature: Feature) -> Feature:
         """Create a Feature.
 
@@ -6727,6 +7002,7 @@ class FeatureEngineeringAPI:
 
         :returns: :class:`Feature`
         """
+
         body = feature.as_dict()
         headers = {
             "Accept": "application/json",
@@ -6744,6 +7020,7 @@ class FeatureEngineeringAPI:
 
         :returns: :class:`MaterializedFeature`
         """
+
         body = materialized_feature.as_dict()
         headers = {
             "Accept": "application/json",
@@ -6899,6 +7176,7 @@ class FeatureEngineeringAPI:
 
         :returns: :class:`Feature`
         """
+
         body = feature.as_dict()
         query = {}
         if update_mask is not None:
@@ -6928,6 +7206,7 @@ class FeatureEngineeringAPI:
 
         :returns: :class:`MaterializedFeature`
         """
+
         body = materialized_feature.as_dict()
         query = {}
         if update_mask is not None:
@@ -6966,6 +7245,7 @@ class FeatureStoreAPI:
 
         :returns: :class:`OnlineStore`
         """
+
         body = online_store.as_dict()
         headers = {
             "Accept": "application/json",
@@ -7047,6 +7327,7 @@ class FeatureStoreAPI:
 
         :returns: :class:`PublishTableResponse`
         """
+
         body = {}
         if publish_spec is not None:
             body["publish_spec"] = publish_spec.as_dict()
@@ -7072,6 +7353,7 @@ class FeatureStoreAPI:
 
         :returns: :class:`OnlineStore`
         """
+
         body = online_store.as_dict()
         query = {}
         if update_mask is not None:
@@ -7204,6 +7486,7 @@ class ForecastingAPI:
           Long-running operation waiter for :class:`ForecastingExperiment`.
           See :method:wait_get_experiment_forecasting_succeeded for more details.
         """
+
         body = {}
         if custom_weights_column is not None:
             body["custom_weights_column"] = custom_weights_column
@@ -7326,6 +7609,7 @@ class MaterializedFeaturesAPI:
 
         :returns: :class:`FeatureTag`
         """
+
         body = feature_tag.as_dict()
         headers = {
             "Accept": "application/json",
@@ -7464,6 +7748,7 @@ class MaterializedFeaturesAPI:
 
         :returns: :class:`FeatureTag`
         """
+
         body = feature_tag.as_dict()
         query = {}
         if update_mask is not None:
@@ -7521,6 +7806,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`ApproveTransitionRequestResponse`
         """
+
         body = {}
         if archive_existing_versions is not None:
             body["archive_existing_versions"] = archive_existing_versions
@@ -7553,6 +7839,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`CreateCommentResponse`
         """
+
         body = {}
         if comment is not None:
             body["comment"] = comment
@@ -7583,6 +7870,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`CreateModelResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description
@@ -7627,6 +7915,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`CreateModelVersionResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description
@@ -7672,6 +7961,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`CreateTransitionRequestResponse`
         """
+
         body = {}
         if comment is not None:
             body["comment"] = comment
@@ -7750,6 +8040,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`CreateWebhookResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description
@@ -7953,6 +8244,7 @@ class ModelRegistryAPI:
 
         :returns: Iterator over :class:`ModelVersion`
         """
+
         body = {}
         if name is not None:
             body["name"] = name
@@ -8222,6 +8514,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`RejectTransitionRequestResponse`
         """
+
         body = {}
         if comment is not None:
             body["comment"] = comment
@@ -8249,6 +8542,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`RenameModelResponse`
         """
+
         body = {}
         if name is not None:
             body["name"] = name
@@ -8372,6 +8666,7 @@ class ModelRegistryAPI:
 
 
         """
+
         body = {}
         if key is not None:
             body["key"] = key
@@ -8403,6 +8698,7 @@ class ModelRegistryAPI:
 
 
         """
+
         body = {}
         if key is not None:
             body["key"] = key
@@ -8434,6 +8730,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`RegisteredModelPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -8460,6 +8757,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`TestRegistryWebhookResponse`
         """
+
         body = {}
         if event is not None:
             body["event"] = event.value
@@ -8502,6 +8800,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`TransitionStageResponse`
         """
+
         body = {}
         if archive_existing_versions is not None:
             body["archive_existing_versions"] = archive_existing_versions
@@ -8533,6 +8832,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`UpdateCommentResponse`
         """
+
         body = {}
         if comment is not None:
             body["comment"] = comment
@@ -8556,6 +8856,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`UpdateModelResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description
@@ -8583,6 +8884,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`UpdateModelVersionResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description
@@ -8613,6 +8915,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`RegisteredModelPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -8677,6 +8980,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`UpdateWebhookResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description