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
    outs:
      # NOTE должно совпадать с конфигом
      - data/models/model.torch