diff options
Diffstat (limited to 'tensorflow/tools/pip_package/setup.py')
-rw-r--r-- | tensorflow/tools/pip_package/setup.py | 17 |
1 files changed, 6 insertions, 11 deletions
diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py index 07e876df16..b3787c0edc 100644 --- a/tensorflow/tools/pip_package/setup.py +++ b/tensorflow/tools/pip_package/setup.py @@ -108,21 +108,16 @@ class InstallHeaders(Command): # directories for -I install_dir = re.sub('/google/protobuf/src', '', install_dir) - # Copy eigen code into tensorflow/include, - # tensorflow/include/external/eigen_archive/eigen-eigen-<revision>, - # and tensorflow/include/eigen-eigen-<revision>. + # Copy eigen code into tensorflow/include. # A symlink would do, but the wheel file that gets created ignores # symlink within the directory hierarchy. # NOTE(keveman): Figure out how to customize bdist_wheel package so # we can do the symlink. - if re.search(r'(external/eigen_archive/eigen-eigen-\w+)', install_dir): - extra_dirs = [re.sub('/external/eigen_archive', '', install_dir), - re.sub(r'external/eigen_archive/eigen-eigen-\w+', '', - install_dir)] - for extra_dir in extra_dirs: - if not os.path.exists(extra_dir): - self.mkpath(extra_dir) - self.copy_file(header, extra_dir) + if 'external/eigen_archive/' in install_dir: + extra_dir = install_dir.replace('external/eigen_archive', '') + if not os.path.exists(extra_dir): + self.mkpath(extra_dir) + self.copy_file(header, extra_dir) if not os.path.exists(install_dir): self.mkpath(install_dir) |