Skip to content

Commit 0ab772a

Browse files
update cmd var
1 parent 22ae99c commit 0ab772a

File tree

3 files changed

+7
-6
lines changed

3 files changed

+7
-6
lines changed

ads/aqua/model/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@
6868
from ads.common.oci_resource import SEARCH_TYPE, OCIResource
6969
from ads.common.utils import get_console_link
7070
from 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",

ads/aqua/modeldeployment/deployment.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
from ads.common.utils import get_log_links
4646
from 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:

ads/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@
6565
AQUA_DEPLOYMENT_CONTAINER_METADATA_NAME = "deployment-container"
6666
AQUA_FINETUNING_CONTAINER_METADATA_NAME = "finetune-container"
6767
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"
6969
AQUA_DEPLOYMENT_CONTAINER_URI_METADATA_NAME = "deployment-container-uri"
7070
AQUA_DEPLOYMENT_CONTAINER_OVERRIDE_FLAG_METADATA_NAME = "deployment-container-custom"
7171
AQUA_FINETUNING_CONTAINER_OVERRIDE_FLAG_METADATA_NAME = "finetune-container-custom"

0 commit comments

Comments
 (0)