From fefeefa96e675a9dbcbc61df30192b99e528cfee Mon Sep 17 00:00:00 2001
From: mattmcw <matt@sixteenmillimeter.com>
Date: Fri, 30 Jun 2023 14:10:55 -0400
Subject: [PATCH] Log if email script not found. Correctly call
 completed_mail.sh from project root

---
 scripts/claim.sh | 2 ++
 scripts/train.sh | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/scripts/claim.sh b/scripts/claim.sh
index 31b862e..1d0935b 100644
--- a/scripts/claim.sh
+++ b/scripts/claim.sh
@@ -23,6 +23,8 @@ fi
 
 if [ -f ./scripts/claim_mail.sh ]; then
 	bash ./scripts/claim_mail.sh "${JSON}"
+else
+	echo "No email script, not sending."
 fi
 
 bash ./scripts/train.sh "${JSON}"
diff --git a/scripts/train.sh b/scripts/train.sh
index f3f99c0..d20b929 100644
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -86,6 +86,11 @@ curl -s -X POST \
 	-F "model=@${MODELPATH}" \
 	"${YOLO_WEB_URL}/job/${ID}"
 
+deactivate
+cd ..
+
 if [ -f ./scripts/completed_mail.sh ]; then
 	bash ./scripts/completed_mail.sh "${JSON}"
+else
+	echo "No email script, not sending"
 fi