revert datasets change

Signed-off-by: Brian Dellabetta <[email protected]>

Author: brian-dellabetta

setup.py

@@ -127,8 +127,7 @@ def localversion_func(version: ScmVersion) -> str:
             if BUILD_TYPE == "release"
             else "transformers>=4.53.0"
         ),
-        # datasets v3.6.0 allowed to be compatible with lm_eval
-        ("datasets>=3.6.0,<=4.1.0" if BUILD_TYPE == "release" else "datasets>=3.6.0"),
+        ("datasets>=4.0.0,<=4.1.0" if BUILD_TYPE == "release" else "datasets>=4.0.0"),
         (
             "accelerate>=1.6.0,<=1.10.1"
             if BUILD_TYPE == "release"