diff --git a/.ci/scripts/benchmark_tooling/analyze_benchmark_stability.py b/.ci/scripts/benchmark_tooling/analyze_benchmark_stability.py index d16ecdf4032..d26ceb840f4 100644 --- a/.ci/scripts/benchmark_tooling/analyze_benchmark_stability.py +++ b/.ci/scripts/benchmark_tooling/analyze_benchmark_stability.py @@ -71,6 +71,7 @@ def analyze_latency_stability( # noqa: C901 reference_file=None, output_dir="stability_analysis_results", verbose_level=0, + debug=False, ): print_section_header(f"Analyzing Stability Against Metric '{target_metric}'") print(f"Primary dataset: {primary_file}") @@ -86,6 +87,9 @@ def analyze_latency_stability( # noqa: C901 primary_datasets = {} documents = read_excel_with_json_header(primary_file) + if debug: + print(f"[Debug Mode] Printing documents: {documents}") + for document in documents: sheetName = document.get("sheetName", None) df = document.get("df", None) @@ -1525,6 +1529,13 @@ def main(): default="stability_analysis_results", help="Directory to save analysis results (default: stability_analysis_results)", ) + + parser.add_argument( + "--debug", + action="store_true", + help="Enable debug mode (default: False)", + ) + parser.add_argument( "--verbose-level", type=int, @@ -1542,6 +1553,7 @@ def main(): args.reference_file, args.output_dir, args.verbose_level, + args.debug, ) diff --git a/.ci/scripts/benchmark_tooling/common.py b/.ci/scripts/benchmark_tooling/common.py index 521e9f3b3ce..bc1c4f65975 100644 --- a/.ci/scripts/benchmark_tooling/common.py +++ b/.ci/scripts/benchmark_tooling/common.py @@ -38,13 +38,3 @@ def read_all_csv_with_metadata(folder_path: str) -> List[Dict[str, Any]]: results.append({"groupInfo": meta, "df": df, "sheetName": fname}) print(f"successfully fetched {len(results)} sheets from {folder_path}") return results - - -import logging - -logging.basicConfig(level=logging.INFO) - -# For Excel files (assuming the Excel file is in the current directory) -file_path = "./private.xlsx" -res = read_excel_with_json_header(file_path) -logging.info(res)