mirror of
https://github.com/mozilla/DeepSpeech.git
synced 2025-10-26 11:19:39 +00:00
58 lines
1.7 KiB
Python
Executable File
58 lines
1.7 KiB
Python
Executable File
#! /usr/bin/env python
|
|
|
|
from setuptools import setup, Extension
|
|
from distutils.command.build import build
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
try:
|
|
import numpy
|
|
try:
|
|
numpy_include = numpy.get_include()
|
|
except AttributeError:
|
|
numpy_include = numpy.get_numpy_include()
|
|
except ImportError:
|
|
numpy_include = ''
|
|
assert 'NUMPY_INCLUDE' in os.environ
|
|
|
|
numpy_include = os.getenv('NUMPY_INCLUDE', numpy_include)
|
|
|
|
project_name = 'deepspeech'
|
|
if '--project_name' in sys.argv:
|
|
project_name_idx = sys.argv.index('--project_name')
|
|
project_name = sys.argv[project_name_idx + 1]
|
|
sys.argv.remove('--project_name')
|
|
sys.argv.pop(project_name_idx)
|
|
|
|
class BuildExtFirst(build):
|
|
sub_commands = [('build_ext', build.has_ext_modules),
|
|
('build_py', build.has_pure_modules),
|
|
('build_clib', build.has_c_libraries),
|
|
('build_scripts', build.has_scripts)]
|
|
|
|
model = Extension('_model',
|
|
['python/model.i'],
|
|
include_dirs = [numpy_include],
|
|
libraries = list(map(lambda x: x.strip(), os.getenv('LIBS', '').split('-l')[1:])))
|
|
|
|
utils = Extension('_utils',
|
|
['python/utils.i'],
|
|
include_dirs = [numpy_include],
|
|
libraries = ['deepspeech_utils'])
|
|
|
|
setup(name = project_name,
|
|
description = 'A library for running inference on a DeepSpeech model',
|
|
author = 'Chris Lord',
|
|
author_email='chrislord.net@gmail.com',
|
|
version = '0.0.1',
|
|
package_dir = {'deepspeech': 'python'},
|
|
packages = [ 'deepspeech' ],
|
|
cmdclass = { 'build': BuildExtFirst },
|
|
license = 'MPL-2.0',
|
|
url = 'https://github.com/mozilla/DeepSpeech',
|
|
ext_modules = [model, utils],
|
|
install_requires = [ 'numpy' ],
|
|
include_package_data = True)
|