Skip to content

Commit 3d092ee

Browse files
committed
added the check for Model Provenance again
1 parent 119e5ab commit 3d092ee

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

ads/model/datascience_model.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1537,15 +1537,16 @@ def update(self, **kwargs) -> "DataScienceModel":
15371537
self.dsc_model = self._to_oci_dsc_model(**kwargs).update()
15381538

15391539
logger.debug(f"Updating a model provenance metadata {self.provenance_metadata}")
1540-
try:
1541-
self.dsc_model.get_model_provenance()
1542-
self.dsc_model.update_model_provenance(
1543-
self.provenance_metadata._to_oci_metadata()
1544-
)
1545-
except ModelProvenanceNotFoundError:
1546-
self.dsc_model.create_model_provenance(
1547-
self.provenance_metadata._to_oci_metadata()
1548-
)
1540+
if self.provenance_metadata:
1541+
try:
1542+
self.dsc_model.get_model_provenance()
1543+
self.dsc_model.update_model_provenance(
1544+
self.provenance_metadata._to_oci_metadata()
1545+
)
1546+
except ModelProvenanceNotFoundError:
1547+
self.dsc_model.create_model_provenance(
1548+
self.provenance_metadata._to_oci_metadata()
1549+
)
15491550

15501551
return self.sync()
15511552

0 commit comments

Comments
 (0)