Skip to content
This repository was archived by the owner on May 23, 2024. It is now read-only.

Commit 54b8e82

Browse files
committed
Ensure env var is treated as innt.
1 parent 596a03b commit 54b8e82

File tree

1 file changed

+2
-2
lines changed
  • docker/build_artifacts/sagemaker

1 file changed

+2
-2
lines changed

docker/build_artifacts/sagemaker/serve.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,9 +63,9 @@ def __init__(self):
6363
self._tfs_inter_op_parallelism = os.environ.get("SAGEMAKER_TFS_INTER_OP_PARALLELISM", 0)
6464
self._tfs_intra_op_parallelism = os.environ.get("SAGEMAKER_TFS_INTRA_OP_PARALLELISM", 0)
6565
self._gunicorn_worker_class = os.environ.get("SAGEMAKER_GUNICORN_WORKER_CLASS", "gevent")
66-
self._gunicorn_setup_timeout_seconds = os.environ.get(
66+
self._gunicorn_setup_timeout_seconds = int(os.environ.get(
6767
"SAGEMAKER_GUNICORN_SETUP_TIMEOUT_SECONDS", 30
68-
)
68+
))
6969

7070
if os.environ.get("OMP_NUM_THREADS") is None:
7171
os.environ["OMP_NUM_THREADS"] = "1"

0 commit comments

Comments
 (0)