-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathsetup.py
More file actions
executable file
·173 lines (149 loc) · 5.28 KB
/
setup.py
File metadata and controls
executable file
·173 lines (149 loc) · 5.28 KB
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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
"""A setuptools based setup module.
See:
https://packaging.python.org/en/latest/distributing.html
https://github.com/pypa/sampleproject
"""
# Allow editable install into user site directory.
# See https://github.com/pypa/pip/issues/7953.
import site
import sys
site.ENABLE_USER_SITE = '--user' in sys.argv[1:]
# To use a consistent encoding
from codecs import open
# Always prefer setuptools over distutils
from distutils.core import setup
from os import path
# import numpy
from setuptools import find_packages
# from Cython.Build import cythonize
here = path.abspath(path.dirname(__file__))
# dependencies only required during test
test = [
'pytest',
'pytest-cov',
]
# Get the long description from the relevant file
try:
with open(path.join(here, 'README.md'), encoding='utf-8') as f:
long_description = f.read()
except FileNotFoundError:
long_description = ''
setup(
name='tssep_data',
# Versions should comply with PEP440. For a discussion on single-sourcing
# the version across setup.py and the project code, see
# https://packaging.python.org/en/latest/single_source_version.html
version='0.0.0',
description='TS-SEP: Joint Diarization and Separation Conditioned on Estimated Speaker Embeddings',
long_description=long_description,
# Denotes that our long_description is in Markdown; valid values are
# text/plain, text/x-rst, and text/markdown
#
# Optional if long_description is written in reStructuredText (rst) but
# required for plain-text or Markdown; if unspecified, "applications should
# attempt to render [the long_description] as text/x-rst; charset=UTF-8 and
# fall back to text/plain if it is not valid rst" (see link below)
#
# This field corresponds to the "Description-Content-Type" metadata field:
# https://packaging.python.org/specifications/core-metadata/#description-content-type-optional
long_description_content_type='text/markdown', # Optional (see note above)
# The project's main homepage.
# url='https://github.com/...',
# Author details
author='Christoph Boeddeker',
# author_email='...',
# Choose your license
# license='MIT',
# See https://pypi.python.org/pypi?%3Aaction=list_classifiers
classifiers=[
# How mature is this project? Common values are
# 3 - Alpha
# 4 - Beta
# 5 - Production/Stable
'Development Status :: 3 - Alpha',
# Indicate who your project is intended for
'Intended Audience :: Developers',
'Topic :: Software Development :: Build Tools',
# Pick your license as you wish (should match "license" above)
'License :: OSI Approved :: MIT License',
# Specify the Python versions you support here. In particular, ensure
# that you indicate whether you support Python 2, Python 3 or both.
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
],
# What does your project relate to?
keywords='audio speech',
# You can just specify the packages manually here if your project is
# simple. Or you can use find_packages().
packages=find_packages(exclude=['contrib', 'docs', 'tests*']),
# List run-time dependencies here. These will be installed by pip when
# your project is installed. For an analysis of "install_requires" vs pip's
# requirements files see:
# https://packaging.python.org/en/latest/requirements.html
install_requires=[
'numpy',
'scipy',
'Cython',
'IPython',
'tensorboardX',
'webrtcvad', # LibriCSS dependency
'resampy', # LibriCSS dependency
"dataclasses; python_version<'3.7'", # dataclasses is in py37 buildin
'soundfile',
'einops',
'cached_property',
'lazy_dataset',
'diskcache', # Optional lazy_dataset dependency
'humanfriendly', # Optional lazy_dataset dependency
'nara_wpe',
'kaldi_io',
'fire',
'sacred',
'tqdm',
'editdistance',
'natsort',
'openai-whisper',
'psutil',
'pyroomacoustics',
'pytimeparse',
'questionary',
'pyyaml>=5.1',
'simplejson',
'torch',
'torchvision',
'torchaudio',
'meeteval',
'dlp_mpi',
'paderbox',
'padertorch',
'pb_bss @ git+http://github.com/fgnt/pb_bss',
'rirgen @ git+https://github.com/boeddeker/rirgen',
'sms_wsj @ git+https://github.com/fgnt/sms_wsj',
'mms_msg @ git+https://github.com/fgnt/mms_msg',
'espnet',
# 'espnet_model_zoo',
# 'nemo_toolkit[all]',
],
# Installation problems in a clean, new environment:
# 1. `cython` and `scipy` must be installed manually before using
# `pip install`
# List additional groups of dependencies here (e.g. development
# dependencies). You can install these using the following syntax,
# for example:
# $ pip install -e .[all]
extras_require={
'all': test,
},
# ext_modules=cythonize([
# '...',
# ],
# annotate=True,
# ),
# include_dirs=[numpy.get_include()],
# entry_points={
# "console_scripts": [
# '...',
# ]
# },
)