Skip to content

Commit 1d8c832

Browse files
author
valhassan
committed
Enhance optional dependencies in pyproject.toml by adding torchaudio to cpu and cu128 groups for improved audio processing support.
1 parent 63e3841 commit 1d8c832

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

pyproject.toml

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,11 +40,21 @@ dependencies = [
4040
"PyYAML>=6.0.0,<7.0.0",
4141
"jsonargparse[all]>=4.41.0,<4.42.0",
4242
"jupyterlab>=4.0.0,<5.0.0",
43+
"torch>=2.8.0,<2.9.0 ; extra == 'cpu' or extra == 'cu128'",
4344
]
4445

4546
[project.optional-dependencies]
46-
cpu = ["torch>=2.8.0,<2.9.0", "torchvision>=0.23.0,<0.24.0"]
47-
cu128 = ["torch>=2.8.0,<2.9.0", "torchvision>=0.23.0,<0.24.0"]
47+
cpu = [
48+
"torch>=2.8.0,<2.9.0",
49+
"torchvision>=0.23.0,<0.24.0",
50+
"torchaudio>=2.8.0,<2.9.0",
51+
]
52+
53+
cu128 = [
54+
"torch>=2.8.0,<2.9.0",
55+
"torchvision>=0.23.0,<0.24.0",
56+
"torchaudio>=2.8.0,<2.9.0",
57+
]
4858
dev = ["pytest>=8.0.0,<9.0.0", "ruff>=0.8.0,<1.0.0", "pre-commit>=4.0.0,<5.0.0"]
4959

5060
[project.urls]

0 commit comments

Comments
 (0)