Data pipelines with DVC in MLOps - Cheat Sheet & Quick Revision

Recall & Review

beginner

What is DVC in the context of data pipelines?

DVC (Data Version Control) is a tool that helps manage data, models, and experiments in machine learning projects. It tracks data changes and automates pipelines like code version control.

beginner

How does DVC help in building data pipelines?

DVC lets you define stages in a pipeline with commands and dependencies. It tracks inputs and outputs, so when data or code changes, only necessary steps rerun, saving time and ensuring reproducibility.

intermediate

What is the purpose of the dvc.yaml file?

The dvc.yaml file describes the pipeline stages, their commands, inputs, and outputs. It acts like a recipe that DVC uses to run and reproduce the pipeline steps.

beginner

How do you run a DVC pipeline after defining it?

You run the command dvc repro. This tells DVC to check for changes and rerun only the pipeline stages that need updating.

beginner

Why is versioning data important in machine learning projects?

Versioning data helps track changes, reproduce results, and collaborate safely. It prevents confusion about which data or model version was used for experiments.

What command initializes a new DVC project?

A dvc new

B dvc start

C dvc create

D dvc init

The command dvc init sets up DVC in your project folder.

Which file stores the pipeline stages in DVC?

A dvc.yaml

B pipeline.json

C dvc.lock

D config.yml

The dvc.yaml file defines pipeline stages, commands, inputs, and outputs.

What does the command dvc repro do?

A Removes old data files

B Uploads data to cloud storage

C Runs the pipeline stages that need updating

D Initializes a new pipeline

dvc repro reruns pipeline stages if inputs or code changed.

Why should data be versioned in ML projects?

A To delete old data automatically

B To track changes and reproduce experiments

C To speed up model training

D To encrypt data for security

Versioning data helps keep track of changes and ensures reproducibility.

Which of these is NOT a DVC pipeline component?

A Containers

B Commands

C Stages

D Inputs/Outputs

DVC pipelines use stages, commands, inputs, and outputs, but not containers.

Explain how DVC helps automate and manage data pipelines in machine learning projects.

Think about how DVC tracks inputs, outputs, and commands to run pipelines efficiently.

Describe the role of the dvc.yaml and dvc.lock files in a DVC pipeline.

One file is like a recipe, the other locks the exact ingredients used.

Practice

1. What is the main purpose of using dvc repro in a DVC pipeline?

easy

A. To delete all pipeline data and cache

B. To initialize a new DVC repository

C. To reproduce pipeline stages and update outputs if inputs changed

D. To manually edit pipeline stage commands

Solution

1. Step 1: Understand the role of dvc repro

   This command checks if any inputs or dependencies of pipeline stages have changed.

2. Step 2: Effect of running dvc repro

   If changes are detected, it reruns the affected stages to update outputs accordingly.

3. Final Answer:

   To reproduce pipeline stages and update outputs if inputs changed -> Option C

4. Quick Check:

   dvc repro updates pipeline outputs [OK]

Hint: Remember: repro means rerun changed pipeline parts [OK]

Common Mistakes:

• Confusing repro with initialization commands
• Thinking repro deletes data
• Assuming repro edits pipeline commands

2. Which of the following is the correct syntax to add a pipeline stage with DVC that runs python train.py and outputs model.pkl?

easy

A. dvc stage add -n train -o model.pkl python train.py

B. dvc add stage train -o model.pkl python train.py

C. dvc run -n train -o model.pkl python train.py

D. dvc stage add -n train -d train.py -o model.pkl python train.py

Solution

1. Step 1: Identify required flags for stage creation

   The dvc stage add command requires -n for name, -d for dependencies, and -o for outputs.

2. Step 2: Check which option includes all required flags correctly

   dvc stage add -n train -d train.py -o model.pkl python train.py uses -n train, -d train.py (dependency), and -o model.pkl with the command python train.py.

3. Final Answer:

   dvc stage add -n train -d train.py -o model.pkl python train.py -> Option D

4. Quick Check:

   Stage add needs name, dependency, output flags [OK]

Hint: Stage add needs -n (name), -d (deps), -o (outputs) [OK]

Common Mistakes:

• Omitting the dependency with -d
• Using deprecated dvc run instead of stage add
• Mixing order of flags incorrectly

3. Given this DVC pipeline stage definition in dvc.yaml:

stages:
  preprocess:
    cmd: python preprocess.py data/raw data/processed
    deps:
      - data/raw
      - preprocess.py
    outs:
      - data/processed

What happens when you run dvc repro after modifying data/raw?

medium

A. The preprocess stage reruns and updates data/processed

B. Nothing happens because only preprocess.py changes trigger rerun

C. The pipeline fails due to missing output specification

D. All pipeline stages rerun regardless of changes

Solution

1. Step 1: Identify dependencies of the preprocess stage

   The stage depends on data/raw and preprocess.py.

2. Step 2: Effect of changing data/raw on dvc repro

   Changing a dependency triggers rerun of that stage to update outputs.

3. Final Answer:

   The preprocess stage reruns and updates data/processed -> Option A

4. Quick Check:

   Changed input triggers stage rerun [OK]

Hint: Change in deps triggers rerun of that stage [OK]

Common Mistakes:

• Assuming no rerun if only data changes
• Thinking all stages rerun always
• Confusing outputs with dependencies

4. You run dvc repro but get an error: ERROR: failed to reproduce stage 'train': missing dependency 'data/train.csv'. What is the most likely cause?

medium

A. The file data/train.csv was deleted or moved after pipeline creation

B. The dvc.yaml file is missing the train stage

C. The dvc.lock file is corrupted

D. You forgot to run dvc init before dvc repro

Solution

1. Step 1: Understand the error message

   The error says a dependency file is missing, which means DVC cannot find data/train.csv.

2. Step 2: Common causes of missing dependency errors

   Usually, the file was deleted, renamed, or moved after the pipeline stage was created.

3. Final Answer:

   The file data/train.csv was deleted or moved after pipeline creation -> Option A

4. Quick Check:

   Missing dependency file causes repro error [OK]

Hint: Check if all dependency files exist before repro [OK]

Common Mistakes:

• Assuming dvc.yaml missing stage causes this error
• Blaming dvc.lock corruption without evidence
• Forgetting to initialize repo before repro

5. You want to create a DVC pipeline with two stages: extract that outputs data/raw.csv, and train that depends on data/raw.csv and outputs model.pkl. Which sequence of commands correctly sets up this pipeline?

hard

A. dvc stage add -n train -o model.pkl python train.py

dvc stage add -n extract -d data/raw.csv -o data/raw.csv python extract.py

B. dvc stage add -n extract -o data/raw.csv python extract.py

dvc stage add -n train -d data/raw.csv -o model.pkl python train.py

C. dvc run -n extract -o data/raw.csv python extract.py

dvc run -n train -d data/raw.csv -o model.pkl python train.py

D. dvc add data/raw.csv
dvc add model.pkl

Solution

1. Step 1: Define extract stage with output only

   Extract stage produces data/raw.csv so it needs -n extract and -o data/raw.csv with the command.

2. Step 2: Define train stage depending on extract output

   Train stage depends on data/raw.csv so it needs -d data/raw.csv, outputs model.pkl, and runs python train.py.

3. Step 3: Confirm correct order and commands

   dvc stage add -n extract -o data/raw.csv python extract.py dvc stage add -n train -d data/raw.csv -o model.pkl python train.py correctly adds extract first, then train with proper dependencies and outputs.

4. Final Answer:

   dvc stage add -n extract -o data/raw.csv python extract.py dvc stage add -n train -d data/raw.csv -o model.pkl python train.py -> Option B

5. Quick Check:

   Define stages with correct deps and outputs [OK]

Hint: Add extract stage first, then train with dependency on extract output [OK]

Common Mistakes:

• Adding train stage before extract output exists
• Using dvc add instead of stage add for pipeline steps
• Missing dependencies in train stage