#!/usr/bin/env python3
"""Wrapper for self-contained LTX-2 LoRA training inside a derivation."""

import argparse
import os
import subprocess
from pathlib import Path


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--preprocessed-dir", required=True, type=Path)
    parser.add_argument("--output-dir", required=True, type=Path)
    parser.add_argument("--config", required=True, type=Path)
    parser.add_argument("--ltx2-src", required=True, type=Path)
    args = parser.parse_args()

    env = os.environ.copy()
    subprocess.run([
        "python", "packages/ltx-trainer/scripts/train.py",
        str(args.config),
        "--data-root", str(args.preprocessed_dir),
        "--output-dir", str(args.output_dir),
    ], check=True, env=env, cwd=str(args.ltx2_src))


if __name__ == "__main__":
    main()