From 1e7d993170aa09b3068b85729b092ea015640205 Mon Sep 17 00:00:00 2001
From: RussTreadon-NOAA <Russ.Treadon@noaa.gov>
Date: Sat, 22 Apr 2023 12:09:19 +0000
Subject: [PATCH] add g-w pygw to PYTHONPATH in run_jedi_exe scripts (#357)

---
 test/atm/run_jedi_exe_3denvar.sh | 7 +++++++
 test/atm/run_jedi_exe_3dhofx.sh  | 7 +++++++
 test/atm/run_jedi_exe_3dvar.sh   | 7 +++++++
 test/atm/run_jedi_exe_letkf.sh   | 7 +++++++
 4 files changed, 28 insertions(+)

diff --git a/test/atm/run_jedi_exe_3denvar.sh b/test/atm/run_jedi_exe_3denvar.sh
index 3f716dc42..faa6d09dc 100755
--- a/test/atm/run_jedi_exe_3denvar.sh
+++ b/test/atm/run_jedi_exe_3denvar.sh
@@ -30,6 +30,13 @@ elif [ "$machine" = "orion" ]; then
     gdasfix="/work2/noaa/da/cmartin/GDASApp/fix"
 fi
 
+# Setup python path for workflow utilities and tasks
+export HOMEgfs=$srcdir/../../ # TODO: HOMEgfs had to be hard-coded in config
+echo $HOMEgfs
+pygwPATH="${HOMEgfs}/ush/python:${HOMEgfs}/ush/python/pygw/src"
+PYTHONPATH="${PYTHONPATH:+${PYTHONPATH}:}${pygwPATH}"
+export PYTHONPATH
+
 # Create test run directory
 mkdir -p ${bindir}/test/atm/global-workflow/testrun/gdas_single_test_3denvar
 cd ${bindir}/test/atm/global-workflow/testrun/gdas_single_test_3denvar
diff --git a/test/atm/run_jedi_exe_3dhofx.sh b/test/atm/run_jedi_exe_3dhofx.sh
index 84a9b057a..f55fb6415 100755
--- a/test/atm/run_jedi_exe_3dhofx.sh
+++ b/test/atm/run_jedi_exe_3dhofx.sh
@@ -30,6 +30,13 @@ elif [ "$machine" = "orion" ]; then
     gdasfix="/work2/noaa/da/cmartin/GDASApp/fix"
 fi
 
+# Setup python path for workflow utilities and tasks
+export HOMEgfs=$srcdir/../../ # TODO: HOMEgfs had to be hard-coded in config
+echo $HOMEgfs
+pygwPATH="${HOMEgfs}/ush/python:${HOMEgfs}/ush/python/pygw/src"
+PYTHONPATH="${PYTHONPATH:+${PYTHONPATH}:}${pygwPATH}"
+export PYTHONPATH
+
 # Create test run directory
 mkdir -p ${bindir}/test/atm/global-workflow/testrun/gdas_single_test_hofx3d
 cd ${bindir}/test/atm/global-workflow/testrun/gdas_single_test_hofx3d
diff --git a/test/atm/run_jedi_exe_3dvar.sh b/test/atm/run_jedi_exe_3dvar.sh
index ffc50816e..d333db3cc 100755
--- a/test/atm/run_jedi_exe_3dvar.sh
+++ b/test/atm/run_jedi_exe_3dvar.sh
@@ -30,6 +30,13 @@ elif [ "$machine" = "orion" ]; then
     gdasfix="/work2/noaa/da/cmartin/GDASApp/fix"
 fi
 
+# Setup python path for workflow utilities and tasks
+export HOMEgfs=$srcdir/../../ # TODO: HOMEgfs had to be hard-coded in config
+echo $HOMEgfs
+pygwPATH="${HOMEgfs}/ush/python:${HOMEgfs}/ush/python/pygw/src"
+PYTHONPATH="${PYTHONPATH:+${PYTHONPATH}:}${pygwPATH}"
+export PYTHONPATH
+
 # Create test run directory
 mkdir -p ${bindir}/test/atm/global-workflow/testrun/gdas_single_test_3dvar
 cd ${bindir}/test/atm/global-workflow/testrun/gdas_single_test_3dvar
diff --git a/test/atm/run_jedi_exe_letkf.sh b/test/atm/run_jedi_exe_letkf.sh
index 6039ebcf6..287b67641 100755
--- a/test/atm/run_jedi_exe_letkf.sh
+++ b/test/atm/run_jedi_exe_letkf.sh
@@ -30,6 +30,13 @@ elif [ "$machine" = "orion" ]; then
     gdasfix="/work2/noaa/da/cmartin/GDASApp/fix"
 fi
 
+# Setup python path for workflow utilities and tasks
+export HOMEgfs=$srcdir/../../ # TODO: HOMEgfs had to be hard-coded in config
+echo $HOMEgfs
+pygwPATH="${HOMEgfs}/ush/python:${HOMEgfs}/ush/python/pygw/src"
+PYTHONPATH="${PYTHONPATH:+${PYTHONPATH}:}${pygwPATH}"
+export PYTHONPATH
+
 # Create test run directory
 mkdir -p ${bindir}/test/atm/global-workflow/testrun/gdas_single_test_letkf
 cd ${bindir}/test/atm/global-workflow/testrun/gdas_single_test_letkf