experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update setup.py and setup_with_bazel.py for tflite_runtime package

Browse Source 
There is no runtime dependency on pybind and "has_ext_modules=lambda: True" is needed to generate proper platform-dependent wheel name (e.g. "cp37-cp37m-macosx_10_15_x86_64" instead of "py3-none-any").

PiperOrigin-RevId: 344033512
Change-Id: I1795039d319e1d54715067bc2a876e2c0901ba26
This commit is contained in:
Dmitry Kovalev 2020-11-24 04:50:11 -08:00 committed by TensorFlower Gardener
parent e13c09b7de
commit 6aa31d629f
2 changed files with 3 additions and 2 deletions
tensorflow/lite/tools/pip_package
setup.pysetup_with_bazel.py

2
tensorflow/lite/tools/pip_package/setup.py
View File

@ -209,6 +209,7 @@ setup(
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Topic :: Scientific/Engineering',
'Topic :: Scientific/Engineering :: Mathematics',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
@ -220,7 +221,6 @@ setup(
ext_modules=[ext],
install_requires=[
'numpy >= 1.16.0',
'pybind11 >= 2.4.3',
],
cmdclass={
'build_ext': CustomBuildExt,

3
tensorflow/lite/tools/pip_package/setup_with_bazel.py
View File

@ -42,6 +42,7 @@ setup(
author_email='packages@tensorflow.org',
license='Apache 2.0',
include_package_data=True,
has_ext_modules=lambda: True,
keywords='tflite tensorflow tensor machine learning',
classifiers=[
'Development Status :: 5 - Production/Stable',
@ -54,6 +55,7 @@ setup(
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Topic :: Scientific/Engineering',
'Topic :: Scientific/Engineering :: Mathematics',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
@ -66,5 +68,4 @@ setup(
package_data={'': ['*.so', '*.pyd']},
install_requires=[
'numpy >= 1.16.0',
'pybind11 >= 2.4.3',
])