From df12470e780ca08ff53b096908a71a45fed5e531 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Wed, 21 Sep 2022 14:40:28 +0100
Subject: [PATCH] flip conda and time hopefully we can capture the exit code
 this way

---
 aimodel/slurm-pretrain-plot.job    | 2 +-
 aimodel/slurm-pretrain-predict.job | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/aimodel/slurm-pretrain-plot.job b/aimodel/slurm-pretrain-plot.job
index 1d0c6e9..197f33b 100755
--- a/aimodel/slurm-pretrain-plot.job
+++ b/aimodel/slurm-pretrain-plot.job
@@ -62,6 +62,6 @@ echo ">>> Installing requirements";
 conda run -n py38 pip install -r requirements.txt;
 echo ">>> Training model";
 #shellcheck disable=SC2086
-/usr/bin/env time -v conda run -n py38 src/index.py pretrain-plot -i "${INPUT}" -o "${filepath_output}" ${ARGS};
+conda run -n py38 /usr/bin/env time -v src/index.py pretrain-plot -i "${INPUT}" -o "${filepath_output}" ${ARGS};
 # src/index.py pretrain --input "${INPUT}" --output "${dir_output}" ${ARGS};
 echo ">>> exited with code $?";
diff --git a/aimodel/slurm-pretrain-predict.job b/aimodel/slurm-pretrain-predict.job
index 22f2d04..1ee7d23 100755
--- a/aimodel/slurm-pretrain-predict.job
+++ b/aimodel/slurm-pretrain-predict.job
@@ -69,7 +69,7 @@ echo ">>> Installing requirements";
 conda run -n py38 pip install -r requirements.txt;
 echo ">>> Training model";
 #shellcheck disable=SC2086
-/usr/bin/env time -v conda run -n py38 src/index.py pretrain-predict --input "${INPUT}" --output "${filepath_output}" -c "${CHECKPOINT}"
+conda run -n py38 /usr/bin/env time -v src/index.py pretrain-predict --input "${INPUT}" --output "${filepath_output}" -c "${CHECKPOINT}"
 echo ">>> exited with code $?";
 
 INPUT="${filepath_output}" POSTFIX="${POSTFIX}" sbatch slurm-pretrain-plot.job