Skip to content

Commit

Permalink
pin marshmallow version
Browse files Browse the repository at this point in the history
  • Loading branch information
xiaoyu-work committed Jan 6, 2025
1 parent 521341d commit afd4ff7
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
numpy<2.0
numpy
onnx
optuna
pandas
Expand Down
3 changes: 3 additions & 0 deletions test/requirements-test.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit afd4ff7

Please sign in to comment.