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 6868from ads .common .oci_resource import SEARCH_TYPE , OCIResource
6969from ads .common .utils import get_console_link
7070from ads .config import (
71- AQUA_DEPLOYMENT_CONTAINER_CMD_VAR ,
71+ AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME ,
7272 AQUA_DEPLOYMENT_CONTAINER_METADATA_NAME ,
7373 AQUA_DEPLOYMENT_CONTAINER_URI_METADATA_NAME ,
7474 AQUA_EVALUATION_CONTAINER_METADATA_NAME ,
@@ -715,7 +715,7 @@ def _create_model_catalog_entry(
715715
716716 cmd_vars = generate_tei_cmd_vars (os_path )
717717 metadata .add (
718- key = AQUA_DEPLOYMENT_CONTAINER_CMD_VAR ,
718+ key = AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME ,
719719 value = "," .join (cmd_vars ),
720720 description = f"Inference container cmd vars for { model_name } " ,
721721 category = "Other" ,
Original file line number Diff line number Diff line change 4545from ads .common .utils import get_log_links
4646from ads .config import (
4747 AQUA_CONFIG_FOLDER ,
48+ AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME ,
4849 AQUA_DEPLOYMENT_CONTAINER_METADATA_NAME ,
4950 AQUA_DEPLOYMENT_CONTAINER_URI_METADATA_NAME ,
5051 AQUA_MODEL_DEPLOYMENT_CONFIG ,
@@ -207,7 +208,7 @@ def create(
207208 if not env_var :
208209 env_var = {}
209210 if not cmd_var :
210- cmd_var = {}
211+ cmd_var = []
211212
212213 try :
213214 model_path_prefix = aqua_model .custom_metadata_list .get (
@@ -260,12 +261,12 @@ def create(
260261
261262 try :
262263 cmd_var_string = aqua_model .custom_metadata_list .get (
263- AQUA_DEPLOYMENT_CONTAINER_URI_METADATA_NAME
264+ AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME
264265 ).value
265266 cmd_var .append (cmd_var_string .split ("," ))
266267 except ValueError as err :
267268 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 "
269270 f"field. Please check if the model was registered with { container_type_key } inference container."
270271 ) from err
271272 else :
Original file line number Diff line number Diff line change 6565AQUA_DEPLOYMENT_CONTAINER_METADATA_NAME = "deployment-container"
6666AQUA_FINETUNING_CONTAINER_METADATA_NAME = "finetune-container"
6767AQUA_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"
6969AQUA_DEPLOYMENT_CONTAINER_URI_METADATA_NAME = "deployment-container-uri"
7070AQUA_DEPLOYMENT_CONTAINER_OVERRIDE_FLAG_METADATA_NAME = "deployment-container-custom"
7171AQUA_FINETUNING_CONTAINER_OVERRIDE_FLAG_METADATA_NAME = "finetune-container-custom"
You can’t perform that action at this time.
0 commit comments