Remove response from comparing dfs

sdv-dev · Sep 20, 2024 · bd52d6e · bd52d6e
1 parent de9b4ee
commit bd52d6e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/benchmark/utils.py b/tests/benchmark/utils.py
@@ -108,12 +108,11 @@ def compare_previous_result_with_current():
                         'python_version': python_version,
                     })
 
-    results = {
+    return {
         'unsupported_dtypes': pd.DataFrame(unsupported_dtypes),
         'new_supported_dtypes': pd.DataFrame(new_supported_dtypes),
         'previously_unseen_dtypes': pd.DataFrame(previously_unseen_dtypes),
     }
-    return results, response
 
 
 def save_results_to_json(results, filename=None):
@@ -159,7 +158,7 @@ def calculate_support_percentage(df):
 
 def compare_and_store_results_in_gdrive():
     csv_handler = CSVHandler()
-    comparison_results, response = compare_previous_result_with_current()
+    comparison_results = compare_previous_result_with_current()
 
     results = csv_handler.read('results/')
     sorted_results = {}