mgyigit commited on
Commit
8bb236f
·
verified ·
1 Parent(s): 8974cc0

Update src/saving_utils.py

Browse files
Files changed (1) hide show
  1. src/saving_utils.py +3 -3
src/saving_utils.py CHANGED
@@ -20,7 +20,7 @@ def get_baseline_df(selected_methods, selected_metrics, leaderboard_path="/tmp/l
20
  return leaderboard_df
21
 
22
 
23
- def save_results(method_name, benchmark_types, results, repo_id="HUBioDataLab/probe-data", repo_type="space", temporary=False):
24
  #First, download files to be updated from {repo_id}
25
  download_from_hub(benchmark_types, repo_id, repo_type)
26
 
@@ -42,7 +42,7 @@ def save_results(method_name, benchmark_types, results, repo_id="HUBioDataLab/pr
42
  return 0
43
 
44
 
45
- def download_from_hub(benchmark_types, repo_id="HUBioDataLab/probe-data", repo_type="space"):
46
  api = HfApi(token=os.getenv("api-key")) #load api-key secret
47
 
48
  benchmark_types.append("leaderboard")
@@ -68,7 +68,7 @@ def download_from_hub(benchmark_types, repo_id="HUBioDataLab/probe-data", repo_t
68
  return 0
69
 
70
 
71
- def upload_to_hub(benchmark_types, repo_id="HUBioDataLab/probe-data", repo_type="space"):
72
  api = HfApi(token=os.getenv("api_key")) # Requires authentication via HF_TOKEN
73
 
74
  for benchmark in benchmark_types:
 
20
  return leaderboard_df
21
 
22
 
23
+ def save_results(method_name, benchmark_types, results, repo_id="mgyigit/probe-data", repo_type="space", temporary=False):
24
  #First, download files to be updated from {repo_id}
25
  download_from_hub(benchmark_types, repo_id, repo_type)
26
 
 
42
  return 0
43
 
44
 
45
+ def download_from_hub(benchmark_types, repo_id="mgyigit/probe-data", repo_type="space"):
46
  api = HfApi(token=os.getenv("api-key")) #load api-key secret
47
 
48
  benchmark_types.append("leaderboard")
 
68
  return 0
69
 
70
 
71
+ def upload_to_hub(benchmark_types, repo_id="mgyigit/probe-data", repo_type="space"):
72
  api = HfApi(token=os.getenv("api_key")) # Requires authentication via HF_TOKEN
73
 
74
  for benchmark in benchmark_types: