diff --git a/setup.py b/setup.py index a7cc2a74b5f..ef8263c671d 100644 --- a/setup.py +++ b/setup.py @@ -20,7 +20,7 @@ extras["docs"] = [] extras["test_prod"] = ["pytest", "pytest-xdist", "pytest-subtests", "parameterized"] extras["test_dev"] = [ - "datasets", "evaluate", "transformers", "scipy", "scikit-learn", "deepspeed", "tqdm", "bitsandbytes", "timm" + "datasets", "evaluate", "transformers", "scipy", "scikit-learn", "deepspeed<0.13.0", "tqdm", "bitsandbytes", "timm" ] extras["testing"] = extras["test_prod"] + extras["test_dev"] extras["rich"] = ["rich"]