Spaces:
Runtime error
Runtime error
File size: 1,196 Bytes
ca2da1c 9bd745c ca2da1c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
import argparse
import os
from datasets import load_dataset, Dataset
from huggingface_hub import HfApi
TOKEN = os.environ.get("DEBUG")
api = HfApi(token=TOKEN)
parser = argparse.ArgumentParser()
parser.add_argument(
"--run_dir",
default=None,
type=str,
required=True,
help="Path to the run directory.",
)
parser.add_argument(
"--model_name",
default=None,
type=str,
required=True,
help="Model to benchmark.",
)
args = parser.parse_args()
# Updating request
dataset = load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN).to_pandas()
# Set benchmark to failed
dataset.loc[dataset["model"].isin(args.model_name), ['status']] = "FAILED"
## Read error message
# with open(f"{args.run_dir}/error.log", 'r') as file:
# error_message = file.read()
#
## Add a new column for the error message if necessary
#if "error_message" not in dataset.columns:
# dataset["error_message"] = ""
#dataset.loc[dataset["model"].isin(args.model_name), ['error_message']] = error_message
updated_dataset = Dataset.from_pandas(dataset)
updated_dataset.push_to_hub("EnergyStarAI/requests_debug", split="test", token=TOKEN)
print("Status set to FAILED")
|