From 26a7365dc7b54988477cead98ba3196e477bac57 Mon Sep 17 00:00:00 2001
From: Shailendra Patel <shailendra.patel@cockroachlabs.com>
Date: Mon, 28 Oct 2024 12:06:41 +0530
Subject: [PATCH] drtprod: modify tpch run script.

Modifying tpch run script to fix the number of arguments.

Epic: none

Release note: None
---
 pkg/cmd/drtprod/configs/drt_scale.yaml       | 2 ++
 pkg/cmd/drtprod/scripts/generate_tpch_run.sh | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/pkg/cmd/drtprod/configs/drt_scale.yaml b/pkg/cmd/drtprod/configs/drt_scale.yaml
index 86a146c24d76..adc644ee31c4 100644
--- a/pkg/cmd/drtprod/configs/drt_scale.yaml
+++ b/pkg/cmd/drtprod/configs/drt_scale.yaml
@@ -172,5 +172,7 @@ targets:
         flags:
           scale-factor: 1000
       - script: "pkg/cmd/drtprod/scripts/generate_tpch_run.sh"
+        args:
+          - scale_factor_1000
         flags:
           scale-factor: 1000
diff --git a/pkg/cmd/drtprod/scripts/generate_tpch_run.sh b/pkg/cmd/drtprod/scripts/generate_tpch_run.sh
index 4cc3befdb509..1a13a0e52c21 100755
--- a/pkg/cmd/drtprod/scripts/generate_tpch_run.sh
+++ b/pkg/cmd/drtprod/scripts/generate_tpch_run.sh
@@ -8,7 +8,7 @@
 # This script sets up the tpch run workload script in the workload nodes
 # The --scale-factor and other flags are passed as argument to this script
 # NOTE - This uses CLUSTER and WORKLOAD_CLUSTER environment variable, if not set the script fails
-if [ "$#" -lt 3 ]; then
+if [ "$#" -lt 2 ]; then
   echo "Usage: $0 <script_suffix> <flags to init:--scale-factor,--db>"
   exit 1
 fi