Update src/saving_utils.py

src/saving_utils.py

@@ -6,6 +6,26 @@ from huggingface_hub import HfApi
 script_dir = os.path.dirname(os.path.abspath(__file__))  # Directory of the running script
 
 
+def save_results(method_name, benchmark_types, results, repo_id="mgiyigt/probe-data", repo_type="space"):
+    #First, download files to be updated from {repo_id}
+    download_files_from_hub(benchmark_types, repo_id, repo_type)
+
+    #Update local files
+    for benchmark_type in benchmark_types:
+        if benchmark_type == 'similarity':
+            save_similarity_output(results['similarity'], method_name)
+        elif benchmark_type == 'function':
+            save_function_output(results['function'], method_name)
+        elif benchmark_type == 'family':
+            save_family_output(results['family'], method_name)
+        elif benchmark_type == "affinity":
+            save_affinity_output(results['affinity', method_name])
+
+    #Upload local files to the {repo_id}
+    upload_to_hub(benchmark_types, repo_id, repo_type)
+
+    return 0
+
 
 def download_files_from_hub(benchmark_types, repo_id="mgyigit/probe-data", repo_type="space"):
     api = HfApi(token=os.getenv("api-key")) #load api-key secret