File tree Expand file tree Collapse file tree 3 files changed +7
-6
lines changed Expand file tree Collapse file tree 3 files changed +7
-6
lines changed Original file line number Diff line number Diff line change 68
68
from ads .common .oci_resource import SEARCH_TYPE , OCIResource
69
69
from ads .common .utils import get_console_link
70
70
from ads .config import (
71
- AQUA_DEPLOYMENT_CONTAINER_CMD_VAR ,
71
+ AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME ,
72
72
AQUA_DEPLOYMENT_CONTAINER_METADATA_NAME ,
73
73
AQUA_DEPLOYMENT_CONTAINER_URI_METADATA_NAME ,
74
74
AQUA_EVALUATION_CONTAINER_METADATA_NAME ,
@@ -715,7 +715,7 @@ def _create_model_catalog_entry(
715
715
716
716
cmd_vars = generate_tei_cmd_vars (os_path )
717
717
metadata .add (
718
- key = AQUA_DEPLOYMENT_CONTAINER_CMD_VAR ,
718
+ key = AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME ,
719
719
value = "," .join (cmd_vars ),
720
720
description = f"Inference container cmd vars for { model_name } " ,
721
721
category = "Other" ,
Original file line number Diff line number Diff line change 45
45
from ads .common .utils import get_log_links
46
46
from ads .config import (
47
47
AQUA_CONFIG_FOLDER ,
48
+ AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME ,
48
49
AQUA_DEPLOYMENT_CONTAINER_METADATA_NAME ,
49
50
AQUA_DEPLOYMENT_CONTAINER_URI_METADATA_NAME ,
50
51
AQUA_MODEL_DEPLOYMENT_CONFIG ,
@@ -207,7 +208,7 @@ def create(
207
208
if not env_var :
208
209
env_var = {}
209
210
if not cmd_var :
210
- cmd_var = {}
211
+ cmd_var = []
211
212
212
213
try :
213
214
model_path_prefix = aqua_model .custom_metadata_list .get (
@@ -260,12 +261,12 @@ def create(
260
261
261
262
try :
262
263
cmd_var_string = aqua_model .custom_metadata_list .get (
263
- AQUA_DEPLOYMENT_CONTAINER_URI_METADATA_NAME
264
+ AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME
264
265
).value
265
266
cmd_var .append (cmd_var_string .split ("," ))
266
267
except ValueError as err :
267
268
raise AquaValueError (
268
- f"{ AQUA_DEPLOYMENT_CONTAINER_URI_METADATA_NAME } key is not available in the custom metadata "
269
+ f"{ AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME } key is not available in the custom metadata "
269
270
f"field. Please check if the model was registered with { container_type_key } inference container."
270
271
) from err
271
272
else :
Original file line number Diff line number Diff line change 65
65
AQUA_DEPLOYMENT_CONTAINER_METADATA_NAME = "deployment-container"
66
66
AQUA_FINETUNING_CONTAINER_METADATA_NAME = "finetune-container"
67
67
AQUA_EVALUATION_CONTAINER_METADATA_NAME = "evaluation-container"
68
- AQUA_DEPLOYMENT_CONTAINER_CMD_VAR = "container_cmd_var"
68
+ AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME = "container_cmd_var"
69
69
AQUA_DEPLOYMENT_CONTAINER_URI_METADATA_NAME = "deployment-container-uri"
70
70
AQUA_DEPLOYMENT_CONTAINER_OVERRIDE_FLAG_METADATA_NAME = "deployment-container-custom"
71
71
AQUA_FINETUNING_CONTAINER_OVERRIDE_FLAG_METADATA_NAME = "finetune-container-custom"
You can’t perform that action at this time.
0 commit comments