simple.dvc.yaml

stages:
  stage1:
    cmd: python train.py input output
    deps:
      - input
    outs:
      - output