diff --git a/tests/benchmark/utils.py b/tests/benchmark/utils.py index 3d77a019b..2e3b48af3 100644 --- a/tests/benchmark/utils.py +++ b/tests/benchmark/utils.py @@ -152,11 +152,14 @@ def calculate_support_percentage(df): feature_columns = df.drop(columns=['dtype', 'sdtype']) # Calculate percentage of TRUE values for each row (dtype) percentage_support = feature_columns.mean(axis=1) * 100 - return pd.DataFrame({ + + df = pd.DataFrame({ 'dtype': df['dtype'], 'sdtype': df['sdtype'], 'percentage_supported': percentage_support, }) + df['percentage_supported'] = df['percentage_supported'].round(2) + return df def _version_tuple(version_str): @@ -189,7 +192,7 @@ def compare_and_store_results_in_gdrive(): else: summary[name] = df['percentage_supported'] - summary['average'] = summary[list(results)].mean(axis=1) + summary['average'] = summary[list(results)].mean(axis=1).round(2) for col in summary.columns: if col not in ('sdtype', 'dtype'): summary[col] = summary[col].apply(lambda x: f'{x}%')