File tree Expand file tree Collapse file tree 2 files changed +2
-3
lines changed
tests/unitary/default_setup/model_deployment Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -965,8 +965,7 @@ def _create(
965
965
# adds `--enable_lora` to parameters
966
966
params_dict .update ({"--enable_lora" : UNKNOWN })
967
967
params = build_params_string (params_dict )
968
-
969
- if create_deployment_details .model_name and "--served-model-name" in params :
968
+ elif create_deployment_details .model_name and "--served-model-name" in params :
970
969
# Replace existing --served-model-name argument with custom name provided by user
971
970
params = re .sub (
972
971
r"--served-model-name\s+\S+" ,
Original file line number Diff line number Diff line change @@ -922,7 +922,7 @@ def test_update(
922
922
model_deployment = self .initialize_model_deployment ()
923
923
model_deployment .dsc_model_deployment .id = "test_model_deployment_id"
924
924
update_model_deployment_details = (
925
- model_deployment ._update_model_deployment_details ()
925
+ model_deployment ._update_model_deployment_details (update_type = "ZDT" )
926
926
)
927
927
model_deployment .update (wait_for_completion = True )
928
928
mock_create .assert_called ()
You can’t perform that action at this time.
0 commit comments