Skip to content

Commit 3103bc4

Browse files
authored
Merge pull request #25 from ICIJ/lsummerbell/refactor-deps
refactor[torchcodec]: move torchcodec to optional
2 parents 5775d54 + 7eb53a6 commit 3103bc4

2 files changed

Lines changed: 20 additions & 15 deletions

File tree

pyproject.toml

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,16 +26,17 @@ dependencies = [
2626
"librosa>=0.10.1,<0.12",
2727
"numpy~=2.3",
2828
"pydantic-extra-types[pycountry]~=2.11",
29-
"torchcodec>=0.10.0,<0.12",
3029
"typer~=0.24",
3130
"huggingface-hub~=1.11",
3231
]
3332

3433
[project.optional-dependencies]
34+
preprocessing = [
35+
"torchcodec>=0.10.0,<0.12", # pulls in torch automatically so we've made it optional for a base install
36+
]
3537
nemo = [
3638
"nemo-toolkit[asr]~=2.7",
3739
"torch~=2.10",
38-
"torchaudio~=2.10",
3940
]
4041
pyannote = [
4142
"pyannote-audio~=4.0",
@@ -78,14 +79,16 @@ torchaudio = [
7879
torchcodec = [
7980
{ index = "pytorch-cpu" }
8081
]
82+
fireredasr2s = [
83+
{ git = "https://github.com/FireRedTeam/FireRedASR2S.git" }
84+
]
8185

8286
[dependency-groups]
8387
dev = [
8488
"black>=25.12.0",
8589
"pre-commit>=4.5.1",
8690
"pylint>=4.0.4",
8791
"pytest>=9.0.2",
88-
"fireredasr2s @ git+https://github.com/FireRedTeam/FireRedASR2S.git",
8992
"nest-asyncio>=1.6.0",
9093
"pytest-asyncio>=1.3.0",
9194
]

uv.lock

Lines changed: 14 additions & 12 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)