Update datasets requirement to be looser

Author: krishnan-chandra

pyproject.toml

@@ -32,7 +32,7 @@ dependencies = [
     "regex",
     "ujson",
     "tqdm",
-    "datasets~=2.14.6,<3.0.0",
+    "datasets>=2.14.0,<3.0.0",
     "requests",
     "optuna",
     "pydantic~=2.0",