even safer traps

premAI-io · Oct 23, 2023 · fc96cf8 · fc96cf8
1 parent e7b98ee
commit fc96cf8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cht-petals/setup-petals.sh b/cht-petals/setup-petals.sh
@@ -1,18 +1,20 @@
 #!/usr/bin/env bash
 # Usage: setup-petals.sh [--model-path=<DIR>] [--dht-prefix=<PREFIX>] [--port=<INT>]
+set -eEuo pipefail
 
-tmpdir="${PREM_APPDIR:-.}/$(uuid)"
+tmpdir="${PREM_APPDIR:-.}/petals-$(uuid)"
 
 cleanup(){
   for i in $(jobs -p); do
-    kill -n 9 $i
+    kill -n 9 $i || :
   done
   rm -rf "$tmpdir"
   exit 0
 }
 
 trap "cleanup" SIGTERM
 trap "cleanup" SIGINT
+trap "cleanup" ERR
 
 # clone source
 git clone -n --depth=1 --filter=tree:0 https://github.com/premAI-io/prem-services.git "$tmpdir"