Clean up versioning code for nightly builds.

PiperOrigin-RevId: 249226101

tensorflow_hub/pip_package/build_pip_package.sh

@@ -58,14 +58,6 @@ function main() {
   cp "LICENSE" "${TMPDIR}/LICENSE.txt"
   cp -R "${RUNFILES}/tensorflow_hub" "${TMPDIR}"
 
-  # If we're dealing with a nightly build we need to make sure that the
-  # version changes on a daily basis. This hack edits the setup.py file
-  # accordingly.
-  if [[ "${PROJECT_NAME}" -eq "tf-hub-nightly" ]]; then
-    POSTFIX=".dev$(date +%Y%m%d%H%M)"
-    sed -i -E "s/version=__version__/version=(__version__ + '${POSTFIX}')/" ${TMPDIR}/setup.py
-  fi
-
   pushd ${TMPDIR}
   rm -f MANIFEST
 

tensorflow_hub/pip_package/setup.py

@@ -17,6 +17,7 @@
 from __future__ import division
 from __future__ import print_function
 
+from datetime import datetime
 from setuptools import find_packages
 from setuptools import setup
 
@@ -40,9 +41,15 @@
   sys.argv.remove('--project_name')
   sys.argv.pop(project_name_idx)
 
+# If we're dealing with a nightly build we need to make sure that the
+# version changes for every release.
+version = __version__
+if project_name == 'tf-hub-nightly':
+  version += '.dev' + datetime.now().strftime('%Y%m%d%H%M')
+
 setup(
     name=project_name,  # Automatic: tensorflow_hub, etc. Case insensitive.
-    version=__version__.replace('-', ''),
+    version=version.replace('-', ''),
     description=('TensorFlow Hub is a library to foster the publication, '
                  'discovery, and consumption of reusable parts of machine '
                  'learning models.'),