Skip to content

Commit 3c32175

Browse files
committed
fix based on the comment
1 parent 2f0758f commit 3c32175

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

ads/model/deployment/common/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,14 +128,14 @@ def send_request(
128128

129129
request_kwargs["headers"] = header
130130

131+
request_kwargs["auth"] = header.pop("signer")
131132
if dry_run:
132133
request_kwargs["headers"]["Accept"] = "*/*"
133134
req = requests.Request("POST", endpoint, **request_kwargs).prepare()
134135
if is_json_payload:
135136
return json.loads(req.body)
136137
return req.body
137138
else:
138-
request_kwargs["auth"] = header.pop("signer")
139139
return requests.post(endpoint, **request_kwargs).json()
140140

141141

ads/model/deployment/model_deployment.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1581,8 +1581,8 @@ def _build_model_deployment_configuration_details(self) -> Dict:
15811581
infrastructure.web_concurrency
15821582
)
15831583
runtime.set_spec(runtime.CONST_ENV, environment_variables)
1584-
if hasattr(runtime, "inference_server") and runtime.inference_server and runtime.inference_server.lower() == "triton":
1585-
environment_variables["CONTAINER_TYPE"] = "TRITON"
1584+
if hasattr(runtime, "inference_server") and runtime.inference_server and runtime.inference_server.upper() == MODEL_DEPLOYMENT_INFERENCE_SERVER_TRITON:
1585+
environment_variables["CONTAINER_TYPE"] = MODEL_DEPLOYMENT_INFERENCE_SERVER_TRITON
15861586
runtime.set_spec(runtime.CONST_ENV, environment_variables)
15871587
environment_configuration_details = {
15881588
runtime.CONST_ENVIRONMENT_CONFIG_TYPE: runtime.environment_config_type,

0 commit comments

Comments
 (0)