Дайте своєму проєкту інтерфейс, який ви можете контролювати.

{
  "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