Skip to content

Commit

Permalink
Use TENSORFLOW_VERSION when available during pip_build script (#18739)
Browse files Browse the repository at this point in the history
  • Loading branch information
sampathweb authored Nov 6, 2023
1 parent c497ffc commit 601488f
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions pip_build.py
Original file line number Diff line number Diff line change
Expand Up @@ -434,6 +434,9 @@ def test_wheel(wheel_path, expected_version, requirements_path):
"virtualenv kenv\n"
f"source {os.path.join('kenv', 'bin', 'activate')}\n"
f"pip3 install -r {requirements_path}\n"
"pip3 uninstall -y tensorflow tf-nightly\n"
'pip3 install "${TENSORFLOW_VERSION:-tf-nightly}"\n'
"pip3 uninstall -y keras keras-nightly\n"
f"pip3 install {wheel_path} --force-reinstall\n"
f"python3 -c 'import keras;{checks};print(keras.__version__)'\n"
f"python3 -c 'import tensorflow as tf;tf.compat.v1.layers.Dense'\n"
Expand Down

0 comments on commit 601488f

Please sign in to comment.