File tree Expand file tree Collapse file tree 1 file changed +9
-8
lines changed
src/sagemaker/serve/model_server/multi_model_server Expand file tree Collapse file tree 1 file changed +9
-8
lines changed Original file line number Diff line number Diff line change @@ -130,14 +130,15 @@ def _upload_server_artifacts(
130
130
}
131
131
}
132
132
133
- env_vars = {
134
- "SAGEMAKER_SUBMIT_DIRECTORY" : "/opt/ml/model/code" ,
135
- "SAGEMAKER_PROGRAM" : "inference.py" ,
136
- "SAGEMAKER_SERVE_SECRET_KEY" : secret_key ,
137
- "SAGEMAKER_REGION" : sagemaker_session .boto_region_name ,
138
- "SAGEMAKER_CONTAINER_LOG_LEVEL" : "10" ,
139
- "LOCAL_PYTHON" : platform .python_version (),
140
- }
133
+ if secret_key :
134
+ env_vars = {
135
+ "SAGEMAKER_SUBMIT_DIRECTORY" : "/opt/ml/model/code" ,
136
+ "SAGEMAKER_PROGRAM" : "inference.py" ,
137
+ "SAGEMAKER_SERVE_SECRET_KEY" : secret_key ,
138
+ "SAGEMAKER_REGION" : sagemaker_session .boto_region_name ,
139
+ "SAGEMAKER_CONTAINER_LOG_LEVEL" : "10" ,
140
+ "LOCAL_PYTHON" : platform .python_version (),
141
+ }
141
142
142
143
return model_data , _update_env_vars (env_vars )
143
144
You can’t perform that action at this time.
0 commit comments