From 6aa31d629f5bba886c4cfb76d43bdb07319f109b Mon Sep 17 00:00:00 2001
From: Dmitry Kovalev <dkovalev@google.com>
Date: Tue, 24 Nov 2020 04:50:11 -0800
Subject: [PATCH] Update setup.py and setup_with_bazel.py for tflite_runtime
package
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
---
tensorflow/lite/tools/pip_package/setup.py | 2 +-
tensorflow/lite/tools/pip_package/setup_with_bazel.py | 3 ++-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/tensorflow/lite/tools/pip_package/setup.py b/tensorflow/lite/tools/pip_package/setup.py
index 387f0483850..a85053b1602 100644
--- a/tensorflow/lite/tools/pip_package/setup.py
+++ b/tensorflow/lite/tools/pip_package/setup.py
@@ -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,
diff --git a/tensorflow/lite/tools/pip_package/setup_with_bazel.py b/tensorflow/lite/tools/pip_package/setup_with_bazel.py
index 2c9decc7e55..a451ec14685 100644
--- a/tensorflow/lite/tools/pip_package/setup_with_bazel.py
+++ b/tensorflow/lite/tools/pip_package/setup_with_bazel.py
@@ -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',
])