From 0047e3eab0bda63e3da6735a1c82d730bde0ccbf Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 7 Mar 2021 20:18:30 -0800
Subject: [PATCH] AWS wait && echo "All tasks done." (#2391)

---
 utils/aws/userdata.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/utils/aws/userdata.sh b/utils/aws/userdata.sh
index 36405d1a1565..a6d6e7976cf3 100644
--- a/utils/aws/userdata.sh
+++ b/utils/aws/userdata.sh
@@ -11,7 +11,8 @@ if [ ! -d yolov5 ]; then
   cd yolov5
   bash data/scripts/get_coco.sh && echo "Data done." &
   sudo docker pull ultralytics/yolov5:latest && echo "Docker done." &
-  # python -m pip install --upgrade pip && pip install -r requirements.txt && python detect.py && echo "Requirements done." &
+  python -m pip install --upgrade pip && pip install -r requirements.txt && python detect.py && echo "Requirements done." &
+  wait && echo "All tasks done." # finish background tasks
 else
   echo "Running re-start script." # resume interrupted runs
   i=0