-
Notifications
You must be signed in to change notification settings - Fork 19
/
Copy pathsetup.py
31 lines (26 loc) · 781 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
28
29
30
31
from setuptools import setup
from setuptools.command.install import install
import os
class BuildPackageProtos(install):
def run(self):
install.run(self)
from grpc.tools import command
command.build_package_protos(self.distribution.package_dir[''])
setup(
name='tensorflow_serving_python',
version='0.1',
description='Python client for tensorflow serving',
author="Sebastian Schlecht",
license="MIT",
packages=['tensorflow_serving_python', 'tensorflow_serving_python.protos'],
package_dir={'': 'src'},
setup_requires=['cython'],
install_requires=[
'grpcio', 'grpcio-tools',
'tensorflow',
],
cmdclass={
'install': BuildPackageProtos,
'develop': BuildPackageProtos,
}
)