File size: 476 Bytes
b4f432f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | """CLI entrypoint for the registered benchmark suite."""
from __future__ import annotations
from model.config import ModelConfig
from model.model import SageTransformer
from eval.benchmarks import run_registered_benchmarks
def main() -> None:
"""Run the benchmark registry against a default model instance."""
model = SageTransformer(ModelConfig())
for result in run_registered_benchmarks(model):
print(result)
if __name__ == "__main__":
main()
|