1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
| #!/usr/bin/env python3
|
| """FunASR setup script."""
|
| import os
|
| from setuptools import find_packages
| from setuptools import setup
|
|
| requirements = {
| "install": [
| "setuptools>=38.5.1",
| "humanfriendly",
| "scipy>=1.4.1",
| "librosa",
| "jamo", # For kss
| "PyYAML>=5.1.2",
| "soundfile>=0.12.1",
| "h5py>=2.10.0",
| "kaldiio>=2.17.0",
| "torch_complex",
| "nltk>=3.4.5",
| # ASR
| "sentencepiece",
| "jieba",
| "rotary_embedding_torch",
| # TTS
| "pypinyin>=0.44.0",
| "espnet_tts_frontend",
| # ENH
| "pytorch_wpe",
| "editdistance>=0.5.2",
| "tensorboard",
| "g2p",
| # PAI
| "oss2",
| "edit-distance",
| "textgrid",
| "protobuf",
| ],
| # train: The modules invoked when training only.
| "train": [
| "editdistance",
| "wandb",
| ],
| # all: The modules should be optionally installled due to some reason.
| # Please consider moving them to "install" occasionally
| "all": [
| # NOTE(kamo): Append modules requiring specific pytorch version or torch>1.3.0
| "torch_optimizer",
| "fairscale",
| "transformers",
| ],
| "setup": [
| "numpy",
| "pytest-runner",
| ],
| "test": [
| "pytest>=3.3.0",
| "pytest-timeouts>=1.2.1",
| "pytest-pythonpath>=0.7.3",
| "pytest-cov>=2.7.1",
| "hacking>=2.0.0",
| "mock>=2.0.0",
| "pycodestyle",
| "jsondiff<2.0.0,>=1.2.0",
| "flake8>=3.7.8",
| "flake8-docstrings>=1.3.1",
| "black",
| ],
| "doc": [
| "Jinja2",
| "Sphinx",
| "sphinx-rtd-theme>=0.2.4",
| "sphinx-argparse>=0.2.5",
| "commonmark",
| "recommonmark>=0.4.0",
| "nbsphinx>=0.4.2",
| "sphinx-markdown-tables>=0.0.12",
| "configargparse>=1.2.1"
| ],
| }
| requirements["all"].extend(requirements["train"])
| requirements["test"].extend(requirements["train"])
|
| install_requires = requirements["install"]
| setup_requires = requirements["setup"]
| tests_require = requirements["test"]
| extras_require = {
| k: v for k, v in requirements.items() if k not in ["install", "setup"]
| }
|
| dirname = os.path.dirname(__file__)
| version_file = os.path.join(dirname, "funasr", "version.txt")
| with open(version_file, "r") as f:
| version = f.read().strip()
| setup(
| name="funasr",
| version=version,
| url="https://github.com/alibaba-damo-academy/FunASR.git",
| author="Speech Lab of DAMO Academy, Alibaba Group",
| author_email="funasr@list.alibaba-inc.com",
| description="FunASR: A Fundamental End-to-End Speech Recognition Toolkit",
| long_description=open(os.path.join(dirname, "README.md"), encoding="utf-8").read(),
| long_description_content_type="text/markdown",
| license="The MIT License",
| packages=find_packages(include=["funasr*"]),
| package_data={"funasr": ["version.txt"]},
| install_requires=install_requires,
| setup_requires=setup_requires,
| tests_require=tests_require,
| extras_require=extras_require,
| python_requires=">=3.7.0",
| classifiers=[
| "Programming Language :: Python",
| "Programming Language :: Python :: 3",
| "Programming Language :: Python :: 3.7",
| "Programming Language :: Python :: 3.8",
| "Programming Language :: Python :: 3.9",
| "Development Status :: 5 - Production/Stable",
| "Intended Audience :: Science/Research",
| "Operating System :: POSIX :: Linux",
| "License :: OSI Approved :: Apache Software License",
| "Topic :: Software Development :: Libraries :: Python Modules",
| ],
| )
|
|
