Update pyproject.toml

Signed-off-by: SimJeg <sjegou@nvidia.com>

Author: SimJeg

pyproject.toml

@@ -16,6 +16,7 @@ dependencies = [
     # transformers<4.54 is not supported due to refactoring of the transformers library.
     # transformers 4.54-4.55.2 are not compatible with kvpress due to flash attention bugs in transformers
     "transformers>=4.56",
+    "datasets>=2.21.0,<3",
     "pandas>=2.2.2,<3",
     "accelerate>=1.0.0,<2",
     "requests>=2.32.3,<3",