diff options
| author |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-08 14:06:42 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-08 14:06:42 -0700 |
| commit | 6534589de19fc808ef1c6c4f91c1dacb8523f3a5 (patch) | |
| tree | bcb2fa43793262ecc3d94a5cd709e47d2974e3c0 /tensorflow/tools | |
| parent | eec9ca8f0baccd249a49046fe31b460903e44850 (diff) | |
| parent | 1f1fe5a01af616707b8554d59651fb4925d7faee (diff) | |
Merge pull request #22719 from samikama:fix_pip_package
PiperOrigin-RevId: 216245301
Diffstat (limited to 'tensorflow/tools')
| -rw-r--r-- | tensorflow/tools/pip_package/setup.py | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py index d864a7a039..dd1dca9ee8 100644 --- a/tensorflow/tools/pip_package/setup.py +++ b/tensorflow/tools/pip_package/setup.py @@ -226,13 +226,14 @@ if os.name == 'nt': else: EXTENSION_NAME = 'python/_pywrap_tensorflow_internal.so' -headers = (list(find_files('*.h', 'tensorflow/core')) + - list(find_files('*.h', 'tensorflow/stream_executor')) + - list(find_files('*.h', 'google/protobuf_archive/src')) + - list(find_files('*', 'third_party/eigen3')) + - list(find_files('*.h', - 'tensorflow/include/external/com_google_absl')) + - list(find_files('*', 'tensorflow/include/external/eigen_archive'))) +headers = ( + list(find_files('*.h', 'tensorflow/core')) + list( + find_files('*.h', 'tensorflow/stream_executor')) + + list(find_files('*.h', 'google/protobuf_archive/src')) + list( + find_files('*', 'third_party/eigen3')) + list( + find_files('*.h', 'tensorflow/include/external/com_google_absl')) + + list(find_files('*.inc', 'tensorflow/include/external/com_google_absl')) + + list(find_files('*', 'tensorflow/include/external/eigen_archive'))) setup( name=project_name, |