stages:
  train:
    cmd: poetry run train
    deps:
      - shad_mlops_transformers/trainer.py
      - shad_mlops_transformers/model.py
      - data/raw/arxivData.json
    params:
      - shad_mlops_transformers/config.py:
          - Config.batch_size
          - Config.random_seed
          - Config.test_size
          - Config.epochs
    outs:
      # NOTE должно совпадать с конфигом
      - data/models/model.torch
      - data/raw/mapper.json
      - data/plots/train.png:
          persist: true