minor modification to training workload collection

mitdbg · Nov 14, 2023 · f1977f6 · f1977f6
1 parent fc7d293
commit f1977f6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 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
@@ -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
@@ -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: