List torch as a installation requirements.

This commit is contained in:
Fangjun Kuang 2022-07-11 09:27:45 +08:00
parent e15b2ae6f5
commit c3c415fb73

View File

@ -41,6 +41,7 @@ setuptools.setup(
long_description_content_type="text/markdown", long_description_content_type="text/markdown",
ext_modules=[cmake_extension("_kaldifeat")], ext_modules=[cmake_extension("_kaldifeat")],
cmdclass={"build_ext": BuildExtension, "bdist_wheel": bdist_wheel}, cmdclass={"build_ext": BuildExtension, "bdist_wheel": bdist_wheel},
install_requires=["torch"],
zip_safe=False, zip_safe=False,
classifiers=[ classifiers=[
"Programming Language :: C++", "Programming Language :: C++",