From 29a5bd9851ec5e1fb48ec96b4afd62ecce16e6dc Mon Sep 17 00:00:00 2001
From: elbashandy <a.elbashandy@gmail.com>
Date: Thu, 8 Oct 2020 15:05:19 -0700
Subject: [PATCH] Loading deeplab model using get_teca_assets_root() instead of
 specifying it directly

---
 test/python/CMakeLists.txt               |  3 ---
 test/python/test_deeplabv3p_ar_detect.py | 21 ++++++++++++---------
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/test/python/CMakeLists.txt b/test/python/CMakeLists.txt
index e11652423..43dfd550b 100644
--- a/test/python/CMakeLists.txt
+++ b/test/python/CMakeLists.txt
@@ -375,7 +375,6 @@ teca_add_test(py_test_bayesian_ar_detect_mpi_threads
 
 teca_add_test(py_test_deeplabv3p_ar_detect_serial
     COMMAND  ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test_deeplabv3p_ar_detect.py
-    "${TECA_ASSETS_ROOT}/cascade_deeplab_IVT.pt"
     "${TECA_DATA_ROOT}/ARTMIP_MERRA_2D_2017-05.*\.nc$"
     "${TECA_DATA_ROOT}/test_deeplabv3p_ar_detect" IVT 1
     FEATURES ${TECA_HAS_NETCDF}
@@ -383,7 +382,6 @@ teca_add_test(py_test_deeplabv3p_ar_detect_serial
 
 teca_add_test(py_test_deeplabv3p_ar_detect_threads
     COMMAND  ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test_deeplabv3p_ar_detect.py
-    "${TECA_ASSETS_ROOT}/cascade_deeplab_IVT.pt"
     "${TECA_DATA_ROOT}/ARTMIP_MERRA_2D_2017-05.*\.nc$"
     "${TECA_DATA_ROOT}/test_deeplabv3p_ar_detect" IVT -1
     FEATURES ${TECA_HAS_NETCDF}
@@ -392,7 +390,6 @@ teca_add_test(py_test_deeplabv3p_ar_detect_threads
 teca_add_test(py_test_deeplabv3p_ar_detect_mpi_threads
     COMMAND ${MPIEXEC} -n ${HALF_TEST_CORES} ${PYTHON_EXECUTABLE}
     ${CMAKE_CURRENT_SOURCE_DIR}/test_deeplabv3p_ar_detect.py
-    "${TECA_ASSETS_ROOT}/cascade_deeplab_IVT.pt"
     "${TECA_DATA_ROOT}/ARTMIP_MERRA_2D_2017-05.*\.nc$"
     "${TECA_DATA_ROOT}/test_deeplabv3p_ar_detect" IVT -1
     FEATURES ${TECA_HAS_NETCDF} ${TECA_HAS_MPI} ${MPI4Py_FOUND}
diff --git a/test/python/test_deeplabv3p_ar_detect.py b/test/python/test_deeplabv3p_ar_detect.py
index 2b410bb1b..6a4219916 100644
--- a/test/python/test_deeplabv3p_ar_detect.py
+++ b/test/python/test_deeplabv3p_ar_detect.py
@@ -14,21 +14,24 @@
 set_stack_trace_on_error()
 set_stack_trace_on_mpi_error()
 
-if (len(sys.argv) != 6):
+if (len(sys.argv) != 5):
     sys.stderr.write('\n\nUsage error:\n'
-                     'test_deeplabv3p_ar_detect [deeplab model] '
-                     '[mesh data regex] [baseline] '
-                     '[water vapor var] [num threads]\n\n')
+                     'test_deeplabv3p_ar_detect [mesh data regex] '
+                     '[baseline] [water vapor var] [num threads]\n\n')
     sys.exit(-1)
 
 # parse command line
-deeplab_model = sys.argv[1]
-input_regex = sys.argv[2]
-baseline = sys.argv[3]
-water_vapor_var = sys.argv[4]
-n_threads = int(sys.argv[5])
+input_regex = sys.argv[1]
+baseline = sys.argv[2]
+water_vapor_var = sys.argv[3]
+n_threads = int(sys.argv[4])
 vrb = 1
 
+deeplab_model = os.path.join(
+    get_teca_assets_root(),
+    "cascade_deeplab_IVT.pt"
+    )
+
 cf_reader = teca_cf_reader.New()
 cf_reader.set_files_regex(input_regex)
 cf_reader.set_periodic_in_x(1)