File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -128,14 +128,14 @@ def send_request(
128
128
129
129
request_kwargs ["headers" ] = header
130
130
131
+ request_kwargs ["auth" ] = header .pop ("signer" )
131
132
if dry_run :
132
133
request_kwargs ["headers" ]["Accept" ] = "*/*"
133
134
req = requests .Request ("POST" , endpoint , ** request_kwargs ).prepare ()
134
135
if is_json_payload :
135
136
return json .loads (req .body )
136
137
return req .body
137
138
else :
138
- request_kwargs ["auth" ] = header .pop ("signer" )
139
139
return requests .post (endpoint , ** request_kwargs ).json ()
140
140
141
141
Original file line number Diff line number Diff line change @@ -1581,8 +1581,8 @@ def _build_model_deployment_configuration_details(self) -> Dict:
1581
1581
infrastructure .web_concurrency
1582
1582
)
1583
1583
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
1586
1586
runtime .set_spec (runtime .CONST_ENV , environment_variables )
1587
1587
environment_configuration_details = {
1588
1588
runtime .CONST_ENVIRONMENT_CONFIG_TYPE : runtime .environment_config_type ,
You can’t perform that action at this time.
0 commit comments