dvilasuero commited on
Commit
bc07579
·
verified ·
1 Parent(s): e5c013c

Upload runner.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. runner.py +9 -8
runner.py CHANGED
@@ -8,6 +8,7 @@
8
  # "transformers",
9
  # "accelerate",
10
  # "huggingface_hub",
 
11
  # ]
12
  # ///
13
  """Runner that downloads an eval script and executes it using inspect CLI."""
@@ -58,7 +59,7 @@ def bundle_and_upload_to_space(log_dir: str, hf_space_id: str, hf_token: str):
58
 
59
  if __name__ == "__main__":
60
  if len(sys.argv) < 4:
61
- print("Usage: eval_runner.py <eval_ref> <model> <space_id> [log_dir] [--package-path] [extra_args...]")
62
  sys.exit(1)
63
 
64
  eval_ref = sys.argv[1]
@@ -66,17 +67,17 @@ if __name__ == "__main__":
66
  hf_space_id = sys.argv[3]
67
  log_dir = sys.argv[4] if len(sys.argv) > 4 and not sys.argv[4].startswith("--") else "./logs"
68
 
69
- # Check if it's a package path (e.g., inspect_evals/arc_easy, petri/audit)
70
- is_package_path = "--package-path" in sys.argv
71
- extra_args = [arg for arg in sys.argv[5:] if arg != "--package-path"]
72
 
73
- if is_package_path:
74
- # Package path - use directly with inspect eval
75
- print(f"Using package path: {eval_ref}")
76
  eval_target = eval_ref
77
  cleanup_file = None
78
  else:
79
- # URL - download and use
80
  print(f"Downloading eval from {eval_ref}...")
81
  with urllib.request.urlopen(eval_ref) as response:
82
  eval_code = response.read().decode('utf-8')
 
8
  # "transformers",
9
  # "accelerate",
10
  # "huggingface_hub",
11
+ # "inspect_evals",
12
  # ]
13
  # ///
14
  """Runner that downloads an eval script and executes it using inspect CLI."""
 
59
 
60
  if __name__ == "__main__":
61
  if len(sys.argv) < 4:
62
+ print("Usage: eval_runner.py <eval_ref> <model> <space_id> [log_dir] [--inspect-evals] [extra_args...]")
63
  sys.exit(1)
64
 
65
  eval_ref = sys.argv[1]
 
67
  hf_space_id = sys.argv[3]
68
  log_dir = sys.argv[4] if len(sys.argv) > 4 and not sys.argv[4].startswith("--") else "./logs"
69
 
70
+ # Check if this is an inspect_evals path
71
+ is_inspect_evals = "--inspect-evals" in sys.argv
72
+ extra_args = [arg for arg in sys.argv[5:] if arg != "--inspect-evals"]
73
 
74
+ if is_inspect_evals:
75
+ # Use inspect_evals path directly
76
+ print(f"Using inspect_evals: {eval_ref}")
77
  eval_target = eval_ref
78
  cleanup_file = None
79
  else:
80
+ # Download custom eval script
81
  print(f"Downloading eval from {eval_ref}...")
82
  with urllib.request.urlopen(eval_ref) as response:
83
  eval_code = response.read().decode('utf-8')