stages:
  train:
    cmd: python src/train.py
    deps:
    - data/raw/wildfire-raw-yolov8
    - src/train.py
    - src/utils.py
    - params.yaml
    outs:
    - models/model.pt
    - reports/train_params.yaml:
        cache: false
    metrics:
    - reports/train_metrics.csv:
        cache: false
  test:
    cmd: python src/predict.py
    deps:
      - data/raw/wildfire-raw-yolov8
      - models/model.pt
      - src/predict.py