Skip to content

Commit

Permalink
Remove pinned keras-nightly dependency.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 530319760
  • Loading branch information
jburnim committed May 8, 2023
1 parent 3bc2c32 commit 04a895d
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions testing/dependency_install_lib.sh
Original file line number Diff line number Diff line change
Expand Up @@ -69,9 +69,7 @@ install_tensorflow() {
PIP_FLAGS=${2-}
# NB: tf-nightly pulls in other deps, like numpy, absl, and six, transitively.
TF_VERSION_STR=$(find_good_tf_nightly_version_str $TF_NIGHTLY_PACKAGE)
python -m pip install $PIP_FLAGS \
$TF_NIGHTLY_PACKAGE==$TF_VERSION_STR \
keras-nightly==2.13.0.dev2023032307 # b/275356962
python -m pip install $PIP_FLAGS $TF_NIGHTLY_PACKAGE==$TF_VERSION_STR
}

install_jax() {
Expand Down

0 comments on commit 04a895d

Please sign in to comment.