mgyigit commited on
Commit
b2ad46f
·
verified ·
1 Parent(s): b12fa6d

Update src/saving_utils.py

Browse files
Files changed (1) hide show
  1. src/saving_utils.py +4 -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"):
24
  #First, download files to be updated from {repo_id}
25
  download_from_hub(benchmark_types, repo_id, repo_type)
26
 
@@ -35,8 +35,9 @@ def save_results(method_name, benchmark_types, results, repo_id="HUBioDataLab/pr
35
  elif benchmark_type == "affinity":
36
  save_affinity_output(results['affinity'], method_name)
37
 
38
- #Upload local files to the {repo_id}
39
- upload_to_hub(benchmark_types, repo_id, repo_type)
 
40
 
41
  return 0
42
 
 
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
 
 
35
  elif benchmark_type == "affinity":
36
  save_affinity_output(results['affinity'], method_name)
37
 
38
+ if not temporary:
39
+ #Upload local files to the {repo_id}
40
+ upload_to_hub(benchmark_types, repo_id, repo_type)
41
 
42
  return 0
43