From 1aa31bf6e76af36212b43799a4d6fe33fdbd4777 Mon Sep 17 00:00:00 2001
From: Justin Wozniak <woz@anl.gov>
Date: Fri, 15 Sep 2023 14:05:34 -0500
Subject: [PATCH] Support local non-containerized execution.  closes
 JDACS4C-IMPROVE/Singularity#64

---
 Pilot1/Uno/train.sh | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/Pilot1/Uno/train.sh b/Pilot1/Uno/train.sh
index d1c60b7f..20418b46 100755
--- a/Pilot1/Uno/train.sh
+++ b/Pilot1/Uno/train.sh
@@ -8,7 +8,14 @@ set -eu
 # arg 3 CANDLE_CONFIG
 
 ### Path to your CANDLEized model's main Python script###
-CANDLE_MODEL=/usr/local/Benchmarks/Pilot1/Uno/uno_baseline_keras2.py
+CANDLE_MODEL=uno_baseline_keras2.py
+IMPROVE_MODEL_DIR=${IMPROVE_MODEL_DIR:-$( dirname -- "$0" )}
+CANDLE_MODEL=${IMPROVE_MODEL_DIR}/${CANDLE_MODEL}
+
+if [ ! -f ${CANDLE_MODEL} ] ; then
+  echo No such file ${CANDLE_MODEL}
+  exit 404
+fi
 
 # Make copy of $# before shifts
 ARGC=$#