meg HF Staff commited on
Commit
461665c
·
verified ·
1 Parent(s): c12bfdb

args.model_name --> [args.model_name]

Browse files
Files changed (1) hide show
  1. failed_run.py +1 -1
failed_run.py CHANGED
@@ -30,7 +30,7 @@ args = parser.parse_args()
30
  dataset = load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN).to_pandas()
31
 
32
  # Set benchmark to failed
33
- dataset.loc[dataset["model"].isin(args.model_name), ['status']] = "FAILED"
34
 
35
  try:
36
  # Read error message
 
30
  dataset = load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN).to_pandas()
31
 
32
  # Set benchmark to failed
33
+ dataset.loc[dataset["model"].isin([args.model_name]), ['status']] = "FAILED"
34
 
35
  try:
36
  # Read error message