Skip to content

Commit f28a9d5

Browse files
kylesayrsdsikka
andauthored
[Tests] Add torchcodec dependency for whisper tracing test (#1634)
## Purpose ## * Fix failing tracing test ## Background ## The latest release of `datasets==4.0.0` requires that `torchcodec` be installed in order to use the peoples_speech dataset ## Changes ## * Add `torchcodec` to list of required deps in order to run whisper tracing test * Add `torchcodec` to dev dependencies Signed-off-by: Kyle Sayers <kylesayrs@gmail.com> Co-authored-by: Dipika Sikka <dipikasikka1@gmail.com>
1 parent f53590c commit f28a9d5

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

setup.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,7 @@ def localversion_func(version: ScmVersion) -> str:
146146
"torchvision",
147147
"librosa",
148148
"soundfile",
149+
"torchcodec",
149150
# linting, formatting, and type checking
150151
"black~=24.4.2",
151152
"isort~=5.13.2",

tests/llmcompressor/transformers/tracing/test_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@
121121
WhisperForConditionalGeneration,
122122
["WhisperDecoderLayer"],
123123
"audio",
124-
["librosa", "soundfile"],
124+
["librosa", "soundfile", "torchcodec"],
125125
),
126126
],
127127
)

0 commit comments

Comments
 (0)