@@ -1146,6 +1146,7 @@ def create_saved_dataset(
11461146 storage : SavedDatasetStorage ,
11471147 tags : Optional [Dict [str , str ]] = None ,
11481148 feature_service : Optional [FeatureService ] = None ,
1149+ allow_overwrite : bool = False ,
11491150 ) -> SavedDataset :
11501151 """
11511152 Execute provided retrieval job and persist its outcome in given storage.
@@ -1154,6 +1155,14 @@ def create_saved_dataset(
11541155 Name for the saved dataset should be unique within project, since it's possible to overwrite previously stored dataset
11551156 with the same name.
11561157
1158+ Args:
1159+ from_: The retrieval job whose result should be persisted.
1160+ name: The name of the saved dataset.
1161+ storage: The saved dataset storage object indicating where the result should be persisted.
1162+ tags (optional): A dictionary of key-value pairs to store arbitrary metadata.
1163+ feature_service (optional): The feature service that should be associated with this saved dataset.
1164+ allow_overwrite (optional): If True, the persisted result can overwrite an existing table or file.
1165+
11571166 Returns:
11581167 SavedDataset object with attached RetrievalJob
11591168
@@ -1186,7 +1195,7 @@ def create_saved_dataset(
11861195 dataset .min_event_timestamp = from_ .metadata .min_event_timestamp
11871196 dataset .max_event_timestamp = from_ .metadata .max_event_timestamp
11881197
1189- from_ .persist (storage )
1198+ from_ .persist (storage = storage , allow_overwrite = allow_overwrite )
11901199
11911200 dataset = dataset .with_retrieval_job (
11921201 self ._get_provider ().retrieve_saved_dataset (
0 commit comments