setup.py (tensorflow-2.11.0-rc0) | : | setup.py (tensorflow-2.11.0-rc1) | ||
---|---|---|---|---|
skipping to change at line 48 | skipping to change at line 48 | |||
from setuptools import find_packages | from setuptools import find_packages | |||
from setuptools import setup | from setuptools import setup | |||
from setuptools.command.install import install as InstallCommandBase | from setuptools.command.install import install as InstallCommandBase | |||
from setuptools.dist import Distribution | from setuptools.dist import Distribution | |||
# This version string is semver compatible, but incompatible with pip. | # This version string is semver compatible, but incompatible with pip. | |||
# For pip, we will remove all '-' characters from this string, and use the | # For pip, we will remove all '-' characters from this string, and use the | |||
# result for pip. | # result for pip. | |||
# Also update tensorflow/tensorflow.bzl and | # Also update tensorflow/tensorflow.bzl and | |||
# tensorflow/core/public/version.h | # tensorflow/core/public/version.h | |||
_VERSION = '2.11.0-rc0' | _VERSION = '2.11.0-rc1' | |||
# We use the same setup.py for all tensorflow_* packages and for the nightly | # We use the same setup.py for all tensorflow_* packages and for the nightly | |||
# equivalents (tf_nightly_*). The package is controlled from the argument line | # equivalents (tf_nightly_*). The package is controlled from the argument line | |||
# when building the pip package. | # when building the pip package. | |||
project_name = 'tensorflow' | project_name = 'tensorflow' | |||
if '--project_name' in sys.argv: | if '--project_name' in sys.argv: | |||
project_name_idx = sys.argv.index('--project_name') | project_name_idx = sys.argv.index('--project_name') | |||
project_name = sys.argv[project_name_idx + 1] | project_name = sys.argv[project_name_idx + 1] | |||
sys.argv.remove('--project_name') | sys.argv.remove('--project_name') | |||
sys.argv.pop(project_name_idx) | sys.argv.pop(project_name_idx) | |||
skipping to change at line 119 | skipping to change at line 119 | |||
'grpcio >= 1.24.3, < 2.0' if sys.byteorder == 'little' else None, | 'grpcio >= 1.24.3, < 2.0' if sys.byteorder == 'little' else None, | |||
# TensorFlow exposes the TF API for certain TF ecosystem packages like | # TensorFlow exposes the TF API for certain TF ecosystem packages like | |||
# keras. When TF depends on those packages, the package version needs to | # keras. When TF depends on those packages, the package version needs to | |||
# match the current TF version. For tf_nightly, we install the nightly | # match the current TF version. For tf_nightly, we install the nightly | |||
# variant of each package instead, which must be one version ahead of the | # variant of each package instead, which must be one version ahead of the | |||
# current release version. These also usually have "alpha" or "dev" in their | # current release version. These also usually have "alpha" or "dev" in their | |||
# version name. | # version name. | |||
# These are all updated during the TF release process. | # These are all updated during the TF release process. | |||
standard_or_nightly('tensorboard >= 2.10, < 2.11', | standard_or_nightly('tensorboard >= 2.10, < 2.11', | |||
'tb-nightly ~= 2.11.0.a'), | 'tb-nightly ~= 2.11.0.a'), | |||
standard_or_nightly('tensorflow_estimator >= 2.10.0rc0, < 2.11', | standard_or_nightly('tensorflow_estimator >= 2.11.0rc0, < 2.12', | |||
'tf-estimator-nightly ~= 2.11.0.dev'), | 'tf-estimator-nightly ~= 2.12.0.dev'), | |||
standard_or_nightly('keras >= 2.10.0rc0, < 2.11', | standard_or_nightly('keras >= 2.11.0rc1, < 2.12', | |||
'keras-nightly ~= 2.11.0.dev'), | 'keras-nightly ~= 2.12.0.dev'), | |||
] | ] | |||
REQUIRED_PACKAGES = [p for p in REQUIRED_PACKAGES if p is not None] | REQUIRED_PACKAGES = [p for p in REQUIRED_PACKAGES if p is not None] | |||
if collaborator_build: | if collaborator_build: | |||
# If this is a collaborator build, then build an "installer" wheel and | # If this is a collaborator build, then build an "installer" wheel and | |||
# add the collaborator packages as the only dependencies. | # add the collaborator packages as the only dependencies. | |||
REQUIRED_PACKAGES = [ | REQUIRED_PACKAGES = [ | |||
# Install the TensorFlow package built by AWS if the user is running | # Install the TensorFlow package built by AWS if the user is running | |||
# Linux on an Aarch64 machine. | # Linux on an Aarch64 machine. | |||
standard_or_nightly('tensorflow-cpu-aws', 'tf-nightly-cpu-aws') + '==' + | standard_or_nightly('tensorflow-cpu-aws', 'tf-nightly-cpu-aws') + '==' + | |||
End of changes. 2 change blocks. | ||||
5 lines changed or deleted | 5 lines changed or added |