-
Notifications
You must be signed in to change notification settings - Fork 3
/
setup.py
27 lines (25 loc) · 796 Bytes
/
setup.py
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
from setuptools import setup
from distutils.command.build import build
import subprocess
class Build(build):
def run(self):
# Run the original build command
build.run(self)
# Custom build stuff goes here
protoc_command = ["make", "trans"]
if subprocess.call(protoc_command) != 0:
exit(-1)
build.run(self)
setup(name='neural_transducer',
version='0.1',
description=('python3 version of neural transducer trained with imitation learning '
'(Makarov & Clematide EMNLP 2018)'),
url='',
author='Peter Makarov & Simon Clematide',
author_email='makarov@cl.uzh.ch',
license='MIT',
packages=['trans'],
cmdclass={
'build': Build
},
zip_safe=False)