Skip to content

Commit

Permalink
Add missing requirements for sharktank t5 tests. (#788)
Browse files Browse the repository at this point in the history
Fixes #787. See the workflow
passing now:
https://github.com/nod-ai/shark-ai/actions/runs/12676783895/job/35330696315?pr=788.

We could also install another package that transitively includes
`protobuf`, like `transformers[sentencepiece]` or `onnx`. I'm also not
sure if this should be a test requirement or a release requirement,
given how limited and indirect the usage is, so just starting simple and
explicit.
  • Loading branch information
ScottTodd authored Jan 8, 2025
1 parent 74f03e3 commit 0c53fb0
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions sharktank/requirements-tests.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
datasets==3.0.0
diffusers
parameterized
protobuf
pytest==8.0.0
pytest-html
pytest-xdist==3.5.0

0 comments on commit 0c53fb0

Please sign in to comment.