run.py

from __future__ import annotations

import argparse
import importlib
from typing import Type

from pixelvisionai.core.config import load_config
from pixelvisionai.core.engine import CoreEngine, GameAdapter


def load_adapter(game_name: str) -> Type[GameAdapter]:
    module_path = f"pixelvisionai.games.{game_name}.adapter"
    module = importlib.import_module(module_path)
    adapter_cls = getattr(module, "GameAdapterImpl", None)
    if adapter_cls is None:
        raise AttributeError(f"{module_path} must define GameAdapterImpl")
    if not issubclass(adapter_cls, GameAdapter):
        raise TypeError(f"{module_path}.GameAdapterImpl must inherit GameAdapter")
    return adapter_cls


def main() -> None:
    parser = argparse.ArgumentParser(description="PixelVisionAI runner")
    parser.add_argument("--game", required=True)
    parser.add_argument("--config", required=True)
    parser.add_argument("--max-steps", type=int, default=100)
    args = parser.parse_args()

    config = load_config(args.config)
    adapter = load_adapter(args.game)()
    engine = CoreEngine(adapter=adapter, config=config)
    engine.run(max_steps=args.max_steps)


if __name__ == "__main__":
    main()