@@ -49,7 +49,7 @@ def __init__(self, spec: ForecastOperatorSpec, additional_data: AdditionalData,
49
49
self .test_mode = os .environ .get ("TEST_MODE" , False )
50
50
self .deployment_info = {}
51
51
52
- def _satiny_test (self ):
52
+ def _sanity_test (self ):
53
53
"""
54
54
Function perform sanity test for saved artifact
55
55
"""
@@ -102,7 +102,7 @@ def save_to_catalog(self):
102
102
data_science_env = True )
103
103
104
104
self ._copy_score_file ()
105
- self ._satiny_test ()
105
+ self ._sanity_test ()
106
106
107
107
if isinstance (self .model_obj , dict ):
108
108
series = self .model_obj .keys ()
@@ -119,9 +119,9 @@ def save_to_catalog(self):
119
119
self .catalog_id = catalog_entry .id
120
120
121
121
logger .info (f"Saved { self .model_name } version-v{ self .model_version } to model catalog"
122
- f" with catalog id : { self .catalog_id } " )
122
+ f" with model ocid : { self .catalog_id } " )
123
123
124
- self .deployment_info = {"catalog_id " : self .catalog_id , "series" : list (series )}
124
+ self .deployment_info = {"model_ocid " : self .catalog_id , "series" : list (series )}
125
125
126
126
def create_deployment (self ):
127
127
"""Create a model deployment serving"""
@@ -199,9 +199,10 @@ def create_deployment(self):
199
199
model_deploy_configuration ,
200
200
wait_for_states = [
201
201
"SUCCEEDED" , "FAILED" ])
202
- self .deployment_info ['model_deployment_id ' ] = model_deployment .data .id
202
+ self .deployment_info ['work_request ' ] = model_deployment .data .id
203
203
logger .info (f"deployment metadata :{ model_deployment .data } " )
204
204
md = data_science .get_model_deployment (model_deployment_id = model_deployment .data .resources [0 ].identifier )
205
+ self .deployment_info ['model_deployment_ocid' ] = md .data .id
205
206
endpoint_url = md .data .model_deployment_url
206
207
self .deployment_info ['model_deployment_endpoint' ] = f"{ endpoint_url } /predict"
207
208
0 commit comments