mirrors/kaldifeat
1
0
Fork 0
mirror of https://github.com/csukuangfj/kaldifeat.git synced 2025-08-09 18:12:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Minor fixes.

Browse Source
This commit is contained in:
Fangjun Kuang 2021-08-18 21:57:30 +08:00
parent 60b35c0ed8
commit 5dd4bcb88d
1 changed files with 0 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
setup.py
View File

@ -5,7 +5,6 @@
import re import re
import setuptools import setuptools
import torch
from cmake.cmake_extension import BuildExtension, bdist_wheel, cmake_extension from cmake.cmake_extension import BuildExtension, bdist_wheel, cmake_extension
@ -25,16 +24,6 @@ def get_package_version():
return latest_version return latest_version
def get_pytorch_version():
# if it is 1.7.1+cuda101, then strip +cuda101
return torch.__version__.split("+")[0]
install_requires = [
f"torch=={get_pytorch_version()}",
]
package_name = "kaldifeat" package_name = "kaldifeat"
with open("kaldifeat/python/kaldifeat/__init__.py", "a") as f: with open("kaldifeat/python/kaldifeat/__init__.py", "a") as f:
@ -48,7 +37,6 @@ setuptools.setup(
data_files=[("", ["LICENSE", "README.md"])], data_files=[("", ["LICENSE", "README.md"])],
package_dir={package_name: "kaldifeat/python/kaldifeat"}, package_dir={package_name: "kaldifeat/python/kaldifeat"},
packages=[package_name], packages=[package_name],
install_requires=install_requires,
url="https://github.com/csukuangfj/kaldifeat", url="https://github.com/csukuangfj/kaldifeat",
long_description=read_long_description(), long_description=read_long_description(),
long_description_content_type="text/markdown", long_description_content_type="text/markdown",