Spaces:
Running
Running
File size: 980 Bytes
e317359 | 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 | #!/usr/bin/env python3
"""Build GIFT-Eval-style aggregate tables from leaderboard result files."""
from __future__ import annotations
import argparse
from pathlib import Path
import sys
REPO_ROOT = Path(__file__).resolve().parents[1]
SRC_DIR = REPO_ROOT / "src"
if str(SRC_DIR) not in sys.path:
sys.path.insert(0, str(SRC_DIR))
from tsfm_bench.eval.gift_eval_aggregation import (
DEFAULT_BASELINE_MODEL,
write_aggregated_results,
)
def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--results", type=Path, default=Path("space/results"))
parser.add_argument("--baseline", default=DEFAULT_BASELINE_MODEL)
return parser.parse_args()
def main() -> int:
args = parse_args()
paths = write_aggregated_results(args.results, args.baseline)
for name, path in paths.items():
print(f"{name}: {path}")
return 0
if __name__ == "__main__":
raise SystemExit(main())
|