diff --git a/requirements.txt b/requirements.txt
index cc7070318..786904514 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-numpy<2.0
+numpy
 onnx
 optuna
 pandas
diff --git a/test/requirements-test.txt b/test/requirements-test.txt
index 48a7a9c6c..812a60abb 100644
--- a/test/requirements-test.txt
+++ b/test/requirements-test.txt
@@ -17,6 +17,9 @@ evaluate
 # https://github.com/microsoft/TransformerCompression/issues/183
 # TODO(team): Switch back to the original version once it's compatible with transformers>=4.42.0
 git+https://github.com/xiaoyu-work/TransformerCompression.git ; python_version >= "3.10"
+# latest 3.24.0 will break the pipeline
+# TODO(team): 55399 Switch back to the latest version once it's compatible with the pipeline
+marshmallow<3.24.0
 mlflow>=2.4.0
 neural-compressor
 nncf==2.7.0