pytest-cov==4.1.0
pytest-timeout==2.2.0
pytest-dependency==0.5.1
testbook==0.4.2
requests_mock==1.11.0
# For SwiftStack access
awscli==1.33.33
nbval==0.11.0
# For NvFaidx equivalence tests
pyfaidx==0.8.1.3

# Temporary pin for pytorch-lightning until megatron callbacks in ProgressPrinter can get fixed.
# See https://nvidia.slack.com/archives/C02A7LYGHK8/p1734727482697309
pytorch-lightning<2.5.0
lightning<2.5.0