Skip to content

Commit

Permalink
add expecttest to requirements.txt (#225)
Browse files Browse the repository at this point in the history
* add expecttest to requirements.txt

* update
  • Loading branch information
msaroufim authored May 7, 2024
1 parent f0bdc8f commit cce5960
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
3 changes: 1 addition & 2 deletions dev-requirements.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
# Test utilities
pytest==7.4.0
expecttest
unittest-xml-reporting
parameterized
packaging
Expand All @@ -9,7 +8,7 @@ transformers
# For prototype features and benchmarks
bitsandbytes #needed for testing triton quant / dequant ops for 8-bit optimizers
matplotlib
pandas
pandas

# Custom CUDA Extensions
ninja
2 changes: 2 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,3 +2,5 @@ torch
numpy
sentencepiece
packaging
expecttest # So we can use IS_FBCODE flag
hypothesis # Avoid test derandomization warning

0 comments on commit cce5960

Please sign in to comment.