Fix calling columns

sdv-dev · Oct 2, 2024 · dd211ee · dd211ee
1 parent a55078d
commit dd211ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/benchmark/utils.py b/tests/benchmark/utils.py
@@ -190,7 +190,7 @@ def compare_and_store_results_in_gdrive():
             summary[name] = df['percentage_supported']
 
     summary['average'] = summary[list(results)].mean(axis=1)
-    for col in summary.columns():
+    for col in summary.columns:
         if col not in ('sdtype', 'dtype'):
             summary[col] = summary[col].apply(lambda x: f'{x}%')