h20_context/grpo/scripts/check_grpo_env.py

#!/usr/bin/env python3
from __future__ import annotations

import importlib
import sys


def main() -> int:
    checks = []
    for name in ("trl", "transformers", "peft", "datasets", "accelerate"):
        mod = importlib.import_module(name)
        checks.append((name, getattr(mod, "__version__", "unknown"), getattr(mod, "__file__", "")))

    from trl import GRPOConfig, GRPOTrainer

    print("GRPO environment import check")
    for name, version, path in checks:
        print(f"{name}: {version} ({path})")
    print(f"GRPOTrainer: {GRPOTrainer}")
    print(f"GRPOConfig: {GRPOConfig}")
    print(f"python: {sys.executable}")
    return 0


if __name__ == "__main__":
    raise SystemExit(main())