FilesExpand file tree

 main

/

setup.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

123 lines (104 loc) · 4.27 KB

 main

/

setup.py

Copy path

Top

File metadata and controls

• Code
• Blame

123 lines (104 loc) · 4.27 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

#

# This file is part of wav2vec2_stt.

# (c) Copyright 2021 by David Zurow

# Licensed under the AGPL-3.0; see LICENSE file.

#

from setuptools import find_packages

import datetime, os, re, subprocess

from skbuild import setup

# Force wheel to be platform specific

# https://stackoverflow.com/questions/45150304/how-to-force-a-python-wheel-to-be-platform-specific-when-building-it

# https://github.com/Yelp/dumb-init/blob/48db0c0d0ecb4598d1a6400710445b85d67616bf/setup.py#L11-L27

# https://github.com/google/or-tools/issues/616#issuecomment-371480314

try:

from wheel.bdist_wheel import bdist_wheel as bdist_wheel

class bdist_wheel_impure(bdist_wheel):

def finalize_options(self):

bdist_wheel.finalize_options(self)

# Mark us as not a pure python package

self.root_is_pure = False

def get_tag(self):

python, abi, plat = bdist_wheel.get_tag(self)

# We don't contain any python source

python, abi = 'py2.py3', 'none'

return python, abi, plat

from setuptools.command.install import install

class install_platlib(install):

def finalize_options(self):

install.finalize_options(self)

self.install_lib = self.install_platlib

except ImportError:

bdist_wheel_impure = None

install_platlib = None

here = os.path.abspath(os.path.dirname(__file__))

# https://packaging.python.org/guides/single-sourcing-package-version/

def read(*parts):

with open(os.path.join(here, *parts), 'r') as fp:

return fp.read()

def find_version(*file_paths):

version_file = read(*file_paths)

version_match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]", version_file, re.M)

if version_match:

return version_match.group(1)

raise RuntimeError("Unable to find version string.")

version = find_version('wav2vec2_stt', '__init__.py')

if version.endswith('dev0'):

version = version[:-1] + datetime.datetime.now().strftime('%Y%m%d%H%M%S')

with open(os.path.join(here, 'README.md')) as f:

long_description = f.read()

if not os.path.exists('native/simplectc/CMakeLists.txt'):

subprocess.check_call(['git', 'submodule', 'update', '--init', 'native/simplectc/'])

torch_cmake_prefix_path = os.environ.get('TORCH_CMAKE_PREFIX_PATH') \

or subprocess.check_output(['python', '-c', 'import torch;print(torch.utils.cmake_prefix_path)'], text=True).strip()

setup(

cmdclass={

'bdist_wheel': bdist_wheel_impure,

'install': install_platlib,

},

cmake_args=['-DCMAKE_PREFIX_PATH=' + torch_cmake_prefix_path],

name='wav2vec2_stt',

version=version,

description='Simple Python library, distributed via binary wheels with few direct dependencies, for easily using wav2vec 2.0 models for speech recognition.',

long_description=long_description,

long_description_content_type='text/markdown',

url='https://github.com/daanzu/wav2vec2_stt',

author='David Zurow',

author_email='daanzu@gmail.com',

license='AGPL-3.0',

# For a list of valid classifiers, see https://pypi.org/classifiers/

classifiers=[

# How mature is this project? Common values are

# 3 - Alpha

# 4 - Beta

# 5 - Production/Stable

'Development Status :: 4 - Beta',

'Intended Audience :: Developers',

'License :: OSI Approved :: GNU Affero General Public License v3',

'Programming Language :: Python :: 3',

'Programming Language :: Python :: 3.6',

'Programming Language :: Python :: 3.7',

'Programming Language :: Python :: 3.8',

'Programming Language :: Python :: 3.9',

],

# keywords='speech recognition',

packages=find_packages(exclude=['contrib', 'docs', 'tests']),

python_requires='>=3.6, <4',

install_requires=[

'cffi ~= 1.12',

'numpy ~= 1.16, != 1.19.4',

],

extras_require={

# 'dev': ['check-manifest'],

'test': ['pytest'],

},

package_data={

'wav2vec2_stt': ['libwav2vec2_stt_lib.*'],

'': ['LICENSE'],

},

project_urls={

'Bug Reports': 'https://github.com/daanzu/wav2vec2_stt/issues',

'Funding': 'https://github.com/sponsors/daanzu',

# 'Say Thanks!': 'http://saythanks.io/to/example',

'Source': 'https://github.com/daanzu/wav2vec2_stt/',

},

)