Internal change

PiperOrigin-RevId: 602714202
tensorflow · Jan 30, 2024 · 02e1581 · 02e1581
1 parent 26b66b0
commit 02e1581
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tensorflow_hub/pip_package/build_pip_package.sh b/tensorflow_hub/pip_package/build_pip_package.sh
@@ -58,6 +58,15 @@ function main() {
   cp "tensorflow_hub/LICENSE" "${TMPDIR}/LICENSE.txt"
   cp -R "${RUNFILES}/tensorflow_hub" "${TMPDIR}"
 
+  # Remove this once the library supports Keras 3.
+  # Install tf-keras-nightly instead if tf-nightly is installed.
+  if [[ $TENSORFLOW_VERSION == *tf-nightly ]]
+  then
+    sed -i "s|tf-keras|tf-keras-nightly|g" "${TMPDIR}/setup.py"
+  else
+    echo "Did not replace dependency on tf-keras"
+  fi
+
   pushd ${TMPDIR}
   rm -f MANIFEST