From f1977f659560021ef0d906db15fcc07c971bac8d Mon Sep 17 00:00:00 2001
From: wuziniu <wuziniu2015@gmail.com>
Date: Tue, 14 Nov 2023 11:00:15 -0500
Subject: [PATCH] minor modification to training workload collection

---
 .../benchmark_tools/aurora/database_connection.py               | 2 +-
 workloads/cross_db_benchmark/benchmark_tools/run_workload.py    | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/workloads/cross_db_benchmark/benchmark_tools/aurora/database_connection.py b/workloads/cross_db_benchmark/benchmark_tools/aurora/database_connection.py
index a9f4c396..5d23405e 100644
--- a/workloads/cross_db_benchmark/benchmark_tools/aurora/database_connection.py
+++ b/workloads/cross_db_benchmark/benchmark_tools/aurora/database_connection.py
@@ -247,7 +247,7 @@ def extract_all_data_stats(self, cursor) -> Dict[str, Any]:
         }
 
     def run_query_collect_statistics(
-        self, sql, repetitions=1, prefix="", explain_only=False, plain_run=True
+        self, sql, repetitions=1, prefix="", explain_only=False, plain_run=False
     ):
         analyze_plans = None
         verbose_plan = None
diff --git a/workloads/cross_db_benchmark/benchmark_tools/run_workload.py b/workloads/cross_db_benchmark/benchmark_tools/run_workload.py
index 9eba01ae..c55ced88 100644
--- a/workloads/cross_db_benchmark/benchmark_tools/run_workload.py
+++ b/workloads/cross_db_benchmark/benchmark_tools/run_workload.py
@@ -98,6 +98,7 @@ def run_workload(
             cap_workload=cap_workload,
             rank=rank,
             world_size=world_size,
+            use_boto_client=False,
             s3_output_path=s3_output_path,
         )
     else: