Skip to content

Commit 0a2f117

Browse files
emilyfertigtensorflower-gardener
authored andcommitted
Fix in the script that finds the most recent tf-nightly version.
PiperOrigin-RevId: 451050955
1 parent 9f3f390 commit 0a2f117

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

testing/dependency_install_lib.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ PYTHON_PARSE_PACKAGE_JSON="
2222
import sys
2323
import json
2424
import argparse
25+
import pkg_resources
2526
import sysconfig
2627
2728
@@ -46,7 +47,7 @@ for release, release_info in package_data['releases'].items():
4647
for wheel_info in release_info):
4748
continue
4849
releases.append(release)
49-
print(sorted(releases)[-1])
50+
print(sorted(releases, key=pkg_resources.parse_version)[-1])
5051
"
5152

5253
find_good_tf_nightly_version_str() {

0 commit comments

Comments
 (0)