From 66c8e2fc23cf79d537eb4143da9509ccdb3277cd Mon Sep 17 00:00:00 2001
From: Sai Kiran Polisetty <spolisetty@nvidia.com>
Date: Thu, 14 Sep 2023 14:21:58 +0000
Subject: [PATCH] Update Model paths in Ensemble and BLS L0 scripts

---
 qa/L0_bls_model/test.sh      | 2 +-
 qa/L0_ensemble_model/test.sh | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/qa/L0_bls_model/test.sh b/qa/L0_bls_model/test.sh
index 09bee4962..e1c8db840 100755
--- a/qa/L0_bls_model/test.sh
+++ b/qa/L0_bls_model/test.sh
@@ -20,7 +20,7 @@ create_logs_dir "L0_bls_model"
 # Set test parameters
 MODEL_ANALYZER="$(which model-analyzer)"
 REPO_VERSION=${NVIDIA_TRITON_SERVER_VERSION}
-MODEL_REPOSITORY=${MODEL_REPOSITORY:="/mnt/nvdl/datasets/inferenceserver/$REPO_VERSION/qa_bls_model_repository/"}
+MODEL_REPOSITORY=${MODEL_REPOSITORY:="/opt/triton-model-analyzer/examples/quick-start"}
 QA_MODELS="bls"
 BLS_COMPOSING_MODELS="add"
 INPUT_JSON="$(pwd)/input_data.json"
diff --git a/qa/L0_ensemble_model/test.sh b/qa/L0_ensemble_model/test.sh
index 2ec42e80b..208c58f94 100755
--- a/qa/L0_ensemble_model/test.sh
+++ b/qa/L0_ensemble_model/test.sh
@@ -20,7 +20,7 @@ create_logs_dir "L0_ensemble_model"
 # Set test parameters
 MODEL_ANALYZER="$(which model-analyzer)"
 REPO_VERSION=${NVIDIA_TRITON_SERVER_VERSION}
-MODEL_REPOSITORY=${MODEL_REPOSITORY:="/mnt/nvdl/datasets/inferenceserver/$REPO_VERSION/qa_ensemble_model_repository/qa_simple_ensemble_model_repository"}
+MODEL_REPOSITORY=${MODEL_REPOSITORY:="/opt/triton-model-analyzer/examples/quick-start"}
 QA_MODELS="ensemble_add_sub"
 MODEL_NAMES="$(echo $QA_MODELS | sed 's/ /,/g')"
 TRITON_LAUNCH_MODE=${TRITON_LAUNCH_MODE:="local"}
@@ -33,6 +33,7 @@ FILENAME_SERVER_ONLY="server-metrics.csv"
 FILENAME_INFERENCE_MODEL="model-metrics-inference.csv"
 FILENAME_GPU_MODEL="model-metrics-gpu.csv"
 
+mkdir -p $MODEL_REPOSITORY/ensemble_add_sub/1
 rm -rf $OUTPUT_MODEL_REPOSITORY
 create_result_paths
 SERVER_LOG=$TEST_LOG_DIR/server.log