From afd4ff75a864be2ad60533ff51f0f7379cabb717 Mon Sep 17 00:00:00 2001
From: Xiaoyu <xiaoyuzhang@microsoft.com>
Date: Mon, 6 Jan 2025 21:56:30 +0000
Subject: [PATCH] pin marshmallow version

---
 requirements.txt           | 2 +-
 test/requirements-test.txt | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

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