Skip to content

Commit b9685a2

Browse files
authored
defined tag support while creating model
1 parent 5693bf6 commit b9685a2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ads/aqua/app.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,6 +225,7 @@ def create_model_catalog(
225225
model_taxonomy_metadata: Union[ModelTaxonomyMetadata, Dict],
226226
compartment_id: str,
227227
project_id: str,
228+
defined_tags: Dict = None,
228229
**kwargs,
229230
) -> DataScienceModel:
230231
model = (
@@ -237,7 +238,7 @@ def create_model_catalog(
237238
.with_custom_metadata_list(model_custom_metadata)
238239
.with_defined_metadata_list(model_taxonomy_metadata)
239240
.with_provenance_metadata(ModelProvenanceMetadata(training_id=UNKNOWN))
240-
# TODO: decide what parameters will be needed
241+
.with_defined_tags(**(defined_tags or {})) # Create defined tags when a model is created.
241242
.create(
242243
**kwargs,
243244
)

0 commit comments

Comments
 (0)