プロジェクトに、自分で制御できるインターフェースを。

{
  "schema_version": 1,
  "input": "../data/pilote.jsonl",
  "batch_size": 4,
  "device": "cuda"
}

import argparse
import json
from pathlib import Path

parser = argparse.ArgumentParser(description="プロジェクトの起動を検証する")
parser.add_argument("--config", type=Path, required=True)
parser.add_argument("--run-dir", type=Path, required=True)
args = parser.parse_args()

try:
    config_path = args.config.resolve()
    config = json.loads(config_path.read_text(encoding="utf-8"))
except (OSError, UnicodeError, json.JSONDecodeError) as exc:
    parser.error(f"設定を読み込めません : {exc}")

expected = {"schema_version", "input", "batch_size", "device"}
if not isinstance(config, dict) or set(config) != expected:
    parser.error("必要なフィールド : schema_version, input, batch_size, device")
if type(config["schema_version"]) is not int or config["schema_version"] != 1:
    parser.error("schema_version は 1 である必要があります")
if type(config["batch_size"]) is not int or config["batch_size"] < 1:
    parser.error("batch_size は正の整数である必要があります")
if config["device"] not in ("cpu", "cuda"):
    parser.error("device は cpu または cuda である必要があります")
if not isinstance(config["input"], str) or not config["input"]:
    parser.error("input は空でないパスである必要があります")

input_path = (config_path.parent / config["input"]).resolve()
run_dir = args.run_dir.resolve()
if not input_path.is_file():
    parser.error("入力ファイルがありません")
if run_dir.exists():
    parser.error("新しい出力フォルダを選択してください")

print(json.dumps({
    "status": "configuration_validated",
    "input": str(input_path),
    "run_dir": str(run_dir),
    "device_requested": config["device"],
    "batch_size": config["batch_size"]
}, ensure_ascii=False))

python prepare_run.py --config config/pilote.json --run-dir runs/pilote-001