Skip to content

Commit 8e2a68e

Browse files
BruceChang9688Bruce Zhang
and
Bruce Zhang
authored
feature: Edit test file for supporting TF 2.10 training (#3380)
Co-authored-by: Bruce Zhang <[email protected]>
1 parent f939f89 commit 8e2a68e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integ/test_training_compiler.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def test_tensorflow(
243243
py_version="py39",
244244
git_config={
245245
"repo": "https://github.com/tensorflow/models.git",
246-
"branch": "v2.9.2",
246+
"branch": "v" + ".".join(tensorflow_training_latest_version.split(".")[:2]) + ".0",
247247
},
248248
source_dir=".",
249249
entry_point="official/vision/train.py",

0 commit comments

Comments
 (0)