Skip to content

Commit 651450e

Browse files
Merge branch 'main' into signer_exception_fix
2 parents 6f05a41 + 7cc634c commit 651450e

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

ads/model/datascience_model.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,8 @@ class DataScienceModel(Builder):
115115
Model version set ID
116116
version_label: str
117117
Model version label
118+
version_id: str
119+
Model version id
118120
model_file_description: dict
119121
Contains object path details for models created by reference.
120122
@@ -169,6 +171,8 @@ class DataScienceModel(Builder):
169171
Sets the model version set ID.
170172
with_version_label(self, version_label: str):
171173
Sets the model version label.
174+
with_version_id(self, version_id: str):
175+
Sets the model version id.
172176
with_model_file_description: dict
173177
Sets path details for models created by reference. Input can be either a dict, string or json file and
174178
the schema is dictated by model_file_description_schema.json
@@ -209,6 +213,7 @@ class DataScienceModel(Builder):
209213
CONST_MODEL_VERSION_SET_ID = "modelVersionSetId"
210214
CONST_MODEL_VERSION_SET_NAME = "modelVersionSetName"
211215
CONST_MODEL_VERSION_LABEL = "versionLabel"
216+
CONST_MODEL_VERSION_ID = "versionId"
212217
CONST_TIME_CREATED = "timeCreated"
213218
CONST_LIFECYCLE_STATE = "lifecycleState"
214219
CONST_MODEL_FILE_DESCRIPTION = "modelDescription"
@@ -230,6 +235,7 @@ class DataScienceModel(Builder):
230235
CONST_MODEL_VERSION_SET_ID: "model_version_set_id",
231236
CONST_MODEL_VERSION_SET_NAME: "model_version_set_name",
232237
CONST_MODEL_VERSION_LABEL: "version_label",
238+
CONST_MODEL_VERSION_ID: "version_id",
233239
CONST_TIME_CREATED: "time_created",
234240
CONST_LIFECYCLE_STATE: "lifecycle_state",
235241
CONST_MODEL_FILE_DESCRIPTION: "model_file_description",
@@ -612,6 +618,20 @@ def with_version_label(self, version_label: str):
612618
"""
613619
return self.set_spec(self.CONST_MODEL_VERSION_LABEL, version_label)
614620

621+
@property
622+
def version_id(self) -> str:
623+
return self.get_spec(self.CONST_MODEL_VERSION_ID)
624+
625+
def with_version_id(self, version_id: str):
626+
"""Sets the model version id.
627+
628+
Parameters
629+
----------
630+
version_id: str
631+
The model version id.
632+
"""
633+
return self.set_spec(self.CONST_MODEL_VERSION_ID, version_id)
634+
615635
@property
616636
def model_file_description(self) -> dict:
617637
return self.get_spec(self.CONST_MODEL_FILE_DESCRIPTION)

0 commit comments

Comments
 (0)