Skip to content

Commit 3e0a98a

Browse files
author
Bryannah Hernandez
committed
integ test edits
1 parent 3e133b2 commit 3e0a98a

File tree

1 file changed

+9
-8
lines changed
  • src/sagemaker/serve/model_server/multi_model_server

1 file changed

+9
-8
lines changed

src/sagemaker/serve/model_server/multi_model_server/server.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -130,14 +130,15 @@ def _upload_server_artifacts(
130130
}
131131
}
132132

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+
}
141142

142143
return model_data, _update_env_vars(env_vars)
143144

0 commit comments

Comments
 (0)