Skip to content

Commit 679d44f

Browse files
clean up
1 parent 0ab772a commit 679d44f

File tree

3 files changed

+8
-7
lines changed

3 files changed

+8
-7
lines changed

ads/aqua/common/utils.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1088,7 +1088,7 @@ def list_hf_models(query: str) -> List[str]:
10881088
raise format_hf_custom_error_message(err) from err
10891089

10901090

1091-
def generate_tei_cmd_vars(os_path: str) -> List[str]:
1091+
def generate_tei_cmd_var(os_path: str) -> List[str]:
10921092
"""This utility functions generates CMD params for Text Embedding Inference container. Only the
10931093
essential parameters for OCI model deployment are added, defaults are used for the rest.
10941094
Parameters
@@ -1098,15 +1098,16 @@ def generate_tei_cmd_vars(os_path: str) -> List[str]:
10981098
10991099
Returns
11001100
-------
1101-
List of command line arguments
1101+
cmd_var:
1102+
List of command line arguments
11021103
"""
11031104

11041105
cmd_prefix = "--"
1105-
cmd_vars = [
1106+
cmd_var = [
11061107
cmd_prefix + TextEmbeddingInferenceContainerParams.MODEL_ID,
11071108
f"{AQUA_MODEL_DEPLOYMENT_FOLDER}{ObjectStorageDetails.from_path(os_path.rstrip('/')).filepath}/",
11081109
cmd_prefix + TextEmbeddingInferenceContainerParams.PORT,
11091110
TEI_CONTAINER_DEFAULT_HOST,
11101111
]
11111112

1112-
return cmd_vars
1113+
return cmd_var

ads/aqua/model/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
_build_resource_identifier,
2222
copy_model_config,
2323
create_word_icon,
24-
generate_tei_cmd_vars,
24+
generate_tei_cmd_var,
2525
get_artifact_path,
2626
get_hf_model_info,
2727
list_os_files_with_extension,
@@ -713,7 +713,7 @@ def _create_model_catalog_entry(
713713
category="Other",
714714
)
715715

716-
cmd_vars = generate_tei_cmd_vars(os_path)
716+
cmd_vars = generate_tei_cmd_var(os_path)
717717
metadata.add(
718718
key=AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME,
719719
value=",".join(cmd_vars),

ads/aqua/modeldeployment/deployment.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -263,7 +263,7 @@ def create(
263263
cmd_var_string = aqua_model.custom_metadata_list.get(
264264
AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME
265265
).value
266-
cmd_var.append(cmd_var_string.split(","))
266+
cmd_var.extend(cmd_var_string.split(","))
267267
except ValueError as err:
268268
raise AquaValueError(
269269
f"{AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME} key is not available in the custom metadata "

0 commit comments

Comments
 (0)