--- a +++ b/requirements-test.txt @@ -0,0 +1,15 @@ +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