databricks-sdk 0.69.0__py3-none-any.whl → 0.71.0__py3-none-any.whl

databricks/sdk/service/ml.py

@@ -291,6 +291,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
@@ -1299,11 +1331,16 @@ class Feature:
     description: Optional[str] = None
     """The description of the feature."""
 
+    filter_condition: Optional[str] = None
+    """The filter condition applied to the source data before aggregation."""
+
     def as_dict(self) -> dict:
         """Serializes the Feature into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.description is not None:
             body["description"] = self.description
+        if self.filter_condition is not None:
+            body["filter_condition"] = self.filter_condition
         if self.full_name is not None:
             body["full_name"] = self.full_name
         if self.function:
@@ -1321,6 +1358,8 @@ class Feature:
         body = {}
         if self.description is not None:
             body["description"] = self.description
+        if self.filter_condition is not None:
+            body["filter_condition"] = self.filter_condition
         if self.full_name is not None:
             body["full_name"] = self.full_name
         if self.function:
@@ -1338,6 +1377,7 @@ class Feature:
         """Deserializes the Feature from a dictionary."""
         return cls(
             description=d.get("description", None),
+            filter_condition=d.get("filter_condition", None),
             full_name=d.get("full_name", None),
             function=_from_dict(d, "function", Function),
             inputs=d.get("inputs", None),
@@ -2419,6 +2459,41 @@ class ListFeaturesResponse:
         return cls(features=_repeated_dict(d, "features", Feature), next_page_token=d.get("next_page_token", None))
 
 
+@dataclass
+class ListMaterializedFeaturesResponse:
+    materialized_features: Optional[List[MaterializedFeature]] = None
+    """List of materialized features."""
+
+    next_page_token: Optional[str] = None
+    """Pagination token to request the next page of results for this query."""
+
+    def as_dict(self) -> dict:
+        """Serializes the ListMaterializedFeaturesResponse 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]
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the ListMaterializedFeaturesResponse into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.materialized_features:
+            body["materialized_features"] = self.materialized_features
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> ListMaterializedFeaturesResponse:
+        """Deserializes the ListMaterializedFeaturesResponse from a dictionary."""
+        return cls(
+            materialized_features=_repeated_dict(d, "materialized_features", MaterializedFeature),
+            next_page_token=d.get("next_page_token", None),
+        )
+
+
 @dataclass
 class ListModelsResponse:
     next_page_token: Optional[str] = None
@@ -2937,6 +3012,90 @@ class LoggedModelTag:
         return cls(key=d.get("key", None), value=d.get("value", None))
 
 
+@dataclass
+class MaterializedFeature:
+    """A materialized feature represents a feature that is continuously computed and stored."""
+
+    feature_name: str
+    """The full name of the feature in Unity Catalog."""
+
+    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."""
+
+    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."""
+
+    table_name: Optional[str] = None
+    """The fully qualified Unity Catalog path to the table containing the materialized feature (Delta
+    table or Lakebase table). Output only."""
+
+    def as_dict(self) -> dict:
+        """Serializes the MaterializedFeature into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.feature_name is not None:
+            body["feature_name"] = self.feature_name
+        if self.last_materialization_time is not None:
+            body["last_materialization_time"] = self.last_materialization_time
+        if self.materialized_feature_id is not None:
+            body["materialized_feature_id"] = self.materialized_feature_id
+        if self.offline_store_config:
+            body["offline_store_config"] = self.offline_store_config.as_dict()
+        if self.online_store_config:
+            body["online_store_config"] = self.online_store_config.as_dict()
+        if self.pipeline_schedule_state is not None:
+            body["pipeline_schedule_state"] = self.pipeline_schedule_state.value
+        if self.table_name is not None:
+            body["table_name"] = self.table_name
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the MaterializedFeature into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.feature_name is not None:
+            body["feature_name"] = self.feature_name
+        if self.last_materialization_time is not None:
+            body["last_materialization_time"] = self.last_materialization_time
+        if self.materialized_feature_id is not None:
+            body["materialized_feature_id"] = self.materialized_feature_id
+        if self.offline_store_config:
+            body["offline_store_config"] = self.offline_store_config
+        if self.online_store_config:
+            body["online_store_config"] = self.online_store_config
+        if self.pipeline_schedule_state is not None:
+            body["pipeline_schedule_state"] = self.pipeline_schedule_state
+        if self.table_name is not None:
+            body["table_name"] = self.table_name
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> MaterializedFeature:
+        """Deserializes the MaterializedFeature from a dictionary."""
+        return cls(
+            feature_name=d.get("feature_name", None),
+            last_materialization_time=d.get("last_materialization_time", None),
+            materialized_feature_id=d.get("materialized_feature_id", None),
+            offline_store_config=_from_dict(d, "offline_store_config", OfflineStoreConfig),
+            online_store_config=_from_dict(d, "online_store_config", OnlineStore),
+            pipeline_schedule_state=_enum(d, "pipeline_schedule_state", MaterializedFeaturePipelineScheduleState),
+            table_name=d.get("table_name", None),
+        )
+
+
+class MaterializedFeaturePipelineScheduleState(Enum):
+
+    ACTIVE = "ACTIVE"
+    PAUSED = "PAUSED"
+    SNAPSHOT = "SNAPSHOT"
+
+
 @dataclass
 class Metric:
     """Metric associated with a run, represented as a key-value pair."""
@@ -3613,6 +3772,52 @@ class ModelVersionTag:
         return cls(key=d.get("key", None), value=d.get("value", None))
 
 
+@dataclass
+class OfflineStoreConfig:
+    """Configuration for offline store destination."""
+
+    catalog_name: str
+    """The Unity Catalog catalog name."""
+
+    schema_name: str
+    """The Unity Catalog schema name."""
+
+    table_name_prefix: str
+    """Prefix for Unity Catalog table name. The materialized feature will be stored in a table with
+    this prefix and a generated postfix."""
+
+    def as_dict(self) -> dict:
+        """Serializes the OfflineStoreConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.catalog_name is not None:
+            body["catalog_name"] = self.catalog_name
+        if self.schema_name is not None:
+            body["schema_name"] = self.schema_name
+        if self.table_name_prefix is not None:
+            body["table_name_prefix"] = self.table_name_prefix
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the OfflineStoreConfig into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.catalog_name is not None:
+            body["catalog_name"] = self.catalog_name
+        if self.schema_name is not None:
+            body["schema_name"] = self.schema_name
+        if self.table_name_prefix is not None:
+            body["table_name_prefix"] = self.table_name_prefix
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> OfflineStoreConfig:
+        """Deserializes the OfflineStoreConfig from a dictionary."""
+        return cls(
+            catalog_name=d.get("catalog_name", None),
+            schema_name=d.get("schema_name", None),
+            table_name_prefix=d.get("table_name_prefix", None),
+        )
+
+
 @dataclass
 class OnlineStore:
     """An OnlineStore is a logical database instance that stores and serves features online."""
@@ -4941,6 +5146,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.
@@ -4988,34 +5225,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
@@ -5113,6 +5358,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
@@ -5305,6 +5575,7 @@ class ExperimentsAPI:
 
         :returns: :class:`CreateExperimentResponse`
         """
+
         body = {}
         if artifact_location is not None:
             body["artifact_location"] = artifact_location
@@ -5347,6 +5618,7 @@ class ExperimentsAPI:
 
         :returns: :class:`CreateLoggedModelResponse`
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -5395,6 +5667,7 @@ class ExperimentsAPI:
 
         :returns: :class:`CreateRunResponse`
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -5423,6 +5696,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -5473,6 +5747,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if run_id is not None:
             body["run_id"] = run_id
@@ -5501,6 +5776,7 @@ class ExperimentsAPI:
 
         :returns: :class:`DeleteRunsResponse`
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -5527,6 +5803,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if key is not None:
             body["key"] = key
@@ -5550,6 +5827,7 @@ class ExperimentsAPI:
 
         :returns: :class:`FinalizeLoggedModelResponse`
         """
+
         body = {}
         if status is not None:
             body["status"] = status.value
@@ -5888,6 +6166,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if metrics is not None:
             body["metrics"] = [v.as_dict() for v in metrics]
@@ -5918,6 +6197,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if datasets is not None:
             body["datasets"] = [v.as_dict() for v in datasets]
@@ -5944,6 +6224,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if params is not None:
             body["params"] = [v.as_dict() for v in params]
@@ -5995,6 +6276,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if dataset_digest is not None:
             body["dataset_digest"] = dataset_digest
@@ -6034,6 +6316,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if model_json is not None:
             body["model_json"] = model_json
@@ -6056,6 +6339,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if models is not None:
             body["models"] = [v.as_dict() for v in models]
@@ -6085,6 +6369,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if key is not None:
             body["key"] = key
@@ -6113,6 +6398,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -6133,6 +6419,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if run_id is not None:
             body["run_id"] = run_id
@@ -6161,6 +6448,7 @@ class ExperimentsAPI:
 
         :returns: :class:`RestoreRunsResponse`
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -6202,6 +6490,7 @@ class ExperimentsAPI:
 
         :returns: Iterator over :class:`Experiment`
         """
+
         body = {}
         if filter is not None:
             body["filter"] = filter
@@ -6261,6 +6550,7 @@ class ExperimentsAPI:
 
         :returns: :class:`SearchLoggedModelsResponse`
         """
+
         body = {}
         if datasets is not None:
             body["datasets"] = [v.as_dict() for v in datasets]
@@ -6324,6 +6614,7 @@ class ExperimentsAPI:
 
         :returns: Iterator over :class:`Run`
         """
+
         body = {}
         if experiment_ids is not None:
             body["experiment_ids"] = [v for v in experiment_ids]
@@ -6363,6 +6654,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -6387,6 +6679,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if tags is not None:
             body["tags"] = [v.as_dict() for v in tags]
@@ -6409,6 +6702,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]
@@ -6435,6 +6729,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if key is not None:
             body["key"] = key
@@ -6461,6 +6756,7 @@ class ExperimentsAPI:
 
 
         """
+
         body = {}
         if experiment_id is not None:
             body["experiment_id"] = experiment_id
@@ -6484,6 +6780,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]
@@ -6520,6 +6817,7 @@ class ExperimentsAPI:
 
         :returns: :class:`UpdateRunResponse`
         """
+
         body = {}
         if end_time is not None:
             body["end_time"] = end_time
@@ -6554,6 +6852,7 @@ class FeatureEngineeringAPI:
 
         :returns: :class:`Feature`
         """
+
         body = feature.as_dict()
         headers = {
             "Accept": "application/json",
@@ -6563,6 +6862,24 @@ class FeatureEngineeringAPI:
         res = self._api.do("POST", "/api/2.0/feature-engineering/features", body=body, headers=headers)
         return Feature.from_dict(res)
 
+    def create_materialized_feature(self, materialized_feature: MaterializedFeature) -> MaterializedFeature:
+        """Create a materialized feature.
+
+        :param materialized_feature: :class:`MaterializedFeature`
+          The materialized feature to create.
+
+        :returns: :class:`MaterializedFeature`
+        """
+
+        body = materialized_feature.as_dict()
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
+
+        res = self._api.do("POST", "/api/2.0/feature-engineering/materialized-features", body=body, headers=headers)
+        return MaterializedFeature.from_dict(res)
+
     def delete_feature(self, full_name: str):
         """Delete a Feature.
 
@@ -6578,6 +6895,23 @@ class FeatureEngineeringAPI:
 
         self._api.do("DELETE", f"/api/2.0/feature-engineering/features/{full_name}", headers=headers)
 
+    def delete_materialized_feature(self, materialized_feature_id: str):
+        """Delete a materialized feature.
+
+        :param materialized_feature_id: str
+          The ID of the materialized feature to delete.
+
+
+        """
+
+        headers = {
+            "Accept": "application/json",
+        }
+
+        self._api.do(
+            "DELETE", f"/api/2.0/feature-engineering/materialized-features/{materialized_feature_id}", headers=headers
+        )
+
     def get_feature(self, full_name: str) -> Feature:
         """Get a Feature.
 
@@ -6594,6 +6928,24 @@ class FeatureEngineeringAPI:
         res = self._api.do("GET", f"/api/2.0/feature-engineering/features/{full_name}", headers=headers)
         return Feature.from_dict(res)
 
+    def get_materialized_feature(self, materialized_feature_id: str) -> MaterializedFeature:
+        """Get a materialized feature.
+
+        :param materialized_feature_id: str
+          The ID of the materialized feature.
+
+        :returns: :class:`MaterializedFeature`
+        """
+
+        headers = {
+            "Accept": "application/json",
+        }
+
+        res = self._api.do(
+            "GET", f"/api/2.0/feature-engineering/materialized-features/{materialized_feature_id}", headers=headers
+        )
+        return MaterializedFeature.from_dict(res)
+
     def list_features(self, *, page_size: Optional[int] = None, page_token: Optional[str] = None) -> Iterator[Feature]:
         """List Features.
 
@@ -6623,6 +6975,45 @@ class FeatureEngineeringAPI:
                 return
             query["page_token"] = json["next_page_token"]
 
+    def list_materialized_features(
+        self, *, feature_name: Optional[str] = None, page_size: Optional[int] = None, page_token: Optional[str] = None
+    ) -> Iterator[MaterializedFeature]:
+        """List materialized features.
+
+        :param feature_name: str (optional)
+          Filter by feature name. If specified, only materialized features materialized from this feature will
+          be returned.
+        :param page_size: int (optional)
+          The maximum number of results to return. Defaults to 100 if not specified. Cannot be greater than
+          1000.
+        :param page_token: str (optional)
+          Pagination token to go to the next page based on a previous query.
+
+        :returns: Iterator over :class:`MaterializedFeature`
+        """
+
+        query = {}
+        if feature_name is not None:
+            query["feature_name"] = feature_name
+        if page_size is not None:
+            query["page_size"] = page_size
+        if page_token is not None:
+            query["page_token"] = page_token
+        headers = {
+            "Accept": "application/json",
+        }
+
+        while True:
+            json = self._api.do(
+                "GET", "/api/2.0/feature-engineering/materialized-features", query=query, headers=headers
+            )
+            if "materialized_features" in json:
+                for v in json["materialized_features"]:
+                    yield MaterializedFeature.from_dict(v)
+            if "next_page_token" not in json or not json["next_page_token"]:
+                return
+            query["page_token"] = json["next_page_token"]
+
     def update_feature(self, full_name: str, feature: Feature, update_mask: str) -> Feature:
         """Update a Feature.
 
@@ -6635,6 +7026,7 @@ class FeatureEngineeringAPI:
 
         :returns: :class:`Feature`
         """
+
         body = feature.as_dict()
         query = {}
         if update_mask is not None:
@@ -6649,6 +7041,40 @@ class FeatureEngineeringAPI:
         )
         return Feature.from_dict(res)
 
+    def update_materialized_feature(
+        self, materialized_feature_id: str, materialized_feature: MaterializedFeature, update_mask: str
+    ) -> MaterializedFeature:
+        """Update a materialized feature (pause/resume).
+
+        :param materialized_feature_id: str
+          Unique identifier for the materialized feature.
+        :param materialized_feature: :class:`MaterializedFeature`
+          The materialized feature to update.
+        :param update_mask: str
+          Provide the materialization feature fields which should be updated. Currently, only the
+          pipeline_state field can be updated.
+
+        :returns: :class:`MaterializedFeature`
+        """
+
+        body = materialized_feature.as_dict()
+        query = {}
+        if update_mask is not None:
+            query["update_mask"] = update_mask
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
+
+        res = self._api.do(
+            "PATCH",
+            f"/api/2.0/feature-engineering/materialized-features/{materialized_feature_id}",
+            query=query,
+            body=body,
+            headers=headers,
+        )
+        return MaterializedFeature.from_dict(res)
+
 
 class FeatureStoreAPI:
     """A feature store is a centralized repository that enables data scientists to find and share features. Using
@@ -6669,6 +7095,7 @@ class FeatureStoreAPI:
 
         :returns: :class:`OnlineStore`
         """
+
         body = online_store.as_dict()
         headers = {
             "Accept": "application/json",
@@ -6750,6 +7177,7 @@ class FeatureStoreAPI:
 
         :returns: :class:`PublishTableResponse`
         """
+
         body = {}
         if publish_spec is not None:
             body["publish_spec"] = publish_spec.as_dict()
@@ -6775,6 +7203,7 @@ class FeatureStoreAPI:
 
         :returns: :class:`OnlineStore`
         """
+
         body = online_store.as_dict()
         query = {}
         if update_mask is not None:
@@ -6907,6 +7336,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
@@ -7029,6 +7459,7 @@ class MaterializedFeaturesAPI:
 
         :returns: :class:`FeatureTag`
         """
+
         body = feature_tag.as_dict()
         headers = {
             "Accept": "application/json",
@@ -7167,6 +7598,7 @@ class MaterializedFeaturesAPI:
 
         :returns: :class:`FeatureTag`
         """
+
         body = feature_tag.as_dict()
         query = {}
         if update_mask is not None:
@@ -7224,6 +7656,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`ApproveTransitionRequestResponse`
         """
+
         body = {}
         if archive_existing_versions is not None:
             body["archive_existing_versions"] = archive_existing_versions
@@ -7256,6 +7689,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`CreateCommentResponse`
         """
+
         body = {}
         if comment is not None:
             body["comment"] = comment
@@ -7286,6 +7720,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`CreateModelResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description
@@ -7330,6 +7765,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`CreateModelVersionResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description
@@ -7375,6 +7811,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`CreateTransitionRequestResponse`
         """
+
         body = {}
         if comment is not None:
             body["comment"] = comment
@@ -7453,6 +7890,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`CreateWebhookResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description
@@ -7656,6 +8094,7 @@ class ModelRegistryAPI:
 
         :returns: Iterator over :class:`ModelVersion`
         """
+
         body = {}
         if name is not None:
             body["name"] = name
@@ -7925,6 +8364,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`RejectTransitionRequestResponse`
         """
+
         body = {}
         if comment is not None:
             body["comment"] = comment
@@ -7952,6 +8392,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`RenameModelResponse`
         """
+
         body = {}
         if name is not None:
             body["name"] = name
@@ -8075,6 +8516,7 @@ class ModelRegistryAPI:
 
 
         """
+
         body = {}
         if key is not None:
             body["key"] = key
@@ -8106,6 +8548,7 @@ class ModelRegistryAPI:
 
 
         """
+
         body = {}
         if key is not None:
             body["key"] = key
@@ -8137,6 +8580,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]
@@ -8163,6 +8607,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`TestRegistryWebhookResponse`
         """
+
         body = {}
         if event is not None:
             body["event"] = event.value
@@ -8205,6 +8650,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`TransitionStageResponse`
         """
+
         body = {}
         if archive_existing_versions is not None:
             body["archive_existing_versions"] = archive_existing_versions
@@ -8236,6 +8682,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`UpdateCommentResponse`
         """
+
         body = {}
         if comment is not None:
             body["comment"] = comment
@@ -8259,6 +8706,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`UpdateModelResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description
@@ -8286,6 +8734,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`UpdateModelVersionResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description
@@ -8316,6 +8765,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]
@@ -8380,6 +8830,7 @@ class ModelRegistryAPI:
 
         :returns: :class:`UpdateWebhookResponse`
         """
+
         body = {}
         if description is not None:
             body["description"] = description