diff --git a/pyproject.toml b/pyproject.toml index d1f84dc87..36e9f09ca 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -43,7 +43,7 @@ dependencies = [ ] [project.optional-dependencies] -dev = ["wheel>=0.42.0,<1.0", "packaging>=23.2,<25", "ninja>=1.11.1.1,<2.0", "scikit-learn>=1.0, <2.0", "boto3>=1.34, <2.0"] +dev = ["wheel>=0.42.0,<1.0", "packaging>=23.2,<26", "ninja>=1.11.1.1,<2.0", "scikit-learn>=1.0, <2.0", "boto3>=1.34, <2.0"] flash-attn = ["flash-attn>=2.5.3,<2.8"] aim = ["aim>=3.19.0,<4.0"] mlflow = ["mlflow"]