Bump datasets==4.4.1

constraints.txt

@@ -21,4 +21,5 @@ pytest-xdist==3.5.0
 pytest-forked==1.6.0
 pytest-split==0.9.0
 
+datasets==4.4.1
 matplotlib==3.10.7

examples/llm_compression/onnx/tiny_llama_scale_estimation/requirements.txt

@@ -6,5 +6,5 @@ optimum-onnx==0.0.3
 optimum==2.0.0
 onnx==1.17.0
 onnxruntime==1.21.1
-datasets==2.14.7
+datasets==4.4.1
 pillow==12.0.0

examples/llm_compression/openvino/smollm2_360m_fp8/requirements.txt

@@ -1,4 +1,4 @@
-datasets==4.3.0
+datasets==4.4.1
 openvino==2025.4.0
 optimum-intel[openvino]==1.26.0
 optimum-onnx==0.0.3

examples/llm_compression/openvino/tiny_llama/requirements.txt

@@ -1,4 +1,4 @@
-datasets==2.14.7
+datasets==4.4.1
 onnx==1.17.0
 openvino==2025.4.0
 optimum-intel[openvino]==1.26.0

examples/llm_compression/openvino/tiny_llama_synthetic_data/requirements.txt

@@ -1,5 +1,5 @@
 torch==2.9.0
-datasets==3.0.1
+datasets==4.4.1
 numpy>=1.23.5,<2
 openvino==2025.4.0
 optimum-intel==1.26.0

examples/llm_compression/torch_fx/tiny_llama/requirements.txt

@@ -1,5 +1,5 @@
 transformers==4.53.0
-datasets==2.14.7
+datasets==4.4.1
 openvino==2025.4.0
 optimum==2.0.0
 torch==2.9.0

tests/openvino/requirements.txt

@@ -14,7 +14,7 @@ virtualenv
 addict>=2.4.0
 timm==0.9.2
 efficientnet_pytorch==0.7.1
-datasets==3.0.1
+datasets
 transformers==4.53.0
 optimum-intel==1.26.0
 optimum-onnx==0.0.3

tests/post_training/requirements.txt

@@ -1,6 +1,8 @@
 -c ../../constraints.txt
-# matplotlib is required for the memory monitor tool
-matplotlib
+
+datasets
+matplotlib # required for the memory monitor tool
+
 torch
 torchvision
 onnx
@@ -23,4 +25,3 @@ timm==0.9.2
 accelerate==1.9.0
 transformers==4.53.0
 whowhatbench @ git+https://github.com/openvinotoolkit/[email protected]#subdirectory=tools/who_what_benchmark
-datasets==3.6.0

tests/torch/requirements.txt

@@ -14,7 +14,7 @@ accelerate==1.9.0
 transformers==4.53.0
 
 # Required for movement_sparsity tests
-datasets==3.0.1
+datasets
 evaluate==0.3.0
 openvino
 timm==0.9.2