/requirements-test-ml.txt

# ML framework testing dependencies
# Include core test deps
-r requirements-test-core.txt

# Heavy ML frameworks
torch
torchvision
skorch
datasets

# Additional tools for ML framework tests
pre-commit