From 7e6600a494175e28018ce25f20a975199e90a1cd Mon Sep 17 00:00:00 2001
From: David Andersen <11966194+dtandersen@users.noreply.github.com>
Date: Tue, 5 Mar 2024 13:10:53 -0800
Subject: [PATCH] Update start-workers.sh

---
 start-workers.sh | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/start-workers.sh b/start-workers.sh
index cd3fa28..aa43502 100644
--- a/start-workers.sh
+++ b/start-workers.sh
@@ -7,13 +7,13 @@ if [ -x "$USERSCRIPT" ]; then
     exec "$USERSCRIPT" "$@"
 fi
 
-NUM_WORKERS=2
+NUM_WORKERS=${NUM_WORKERS:-2}
 
-WORKER_CPU_REQUEST=8
-WORKER_CPU_LIMIT=8
-WORKER_MEM_REQUEST=16384M
-WORKER_MEM_LIMIT=16384M
-WORKER_GPU_COUNT=0
+WORKER_CPU_REQUEST=${WORKER_CPU_REQUEST:-8}
+WORKER_CPU_LIMIT=${WORKER_CPU_LIMIT:-8}
+WORKER_MEM_REQUEST=${WORKER_MEM_REQUEST:-16384M}
+WORKER_MEM_LIMIT=${WORKER_MEM_LIMIT:-16384M}
+WORKER_GPU_COUNT=${WORKER_GPU_COUNT:-0}
 
 IMAGE=${JUPYTER_IMAGE_SPEC:-${DOCKER_IMAGE}}
 echo "STARTING WORKERS WITH WORKER_CPU_REQUEST=${WORKER_CPU_REQUEST}"