Spaces:
Sleeping
Sleeping
| #!/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()) | |