added torchaudio requirement to match pytorch docs.

MPoL-dev · Dec 15, 2022 · 9a2c52c · 9a2c52c
1 parent 35f4409
commit 9a2c52c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -65,7 +65,7 @@ def get_version(rel_path):
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/iancze/MPoL",
-    install_requires=["numpy", "scipy", "torch>=1.8.0", "torchvision"],
+    install_requires=["numpy", "scipy", "torch>=1.8.0", "torchvision", "torchaudio"],
     extras_require=EXTRA_REQUIRES,
     packages=setuptools.find_packages("src"),
     package_dir={"": "src"},