Skip to content

Commit c0f7814

Browse files
BruceChang9688Bruce Zhang
authored andcommitted
feature: Edit test file for supporting TF 2.10 training (aws#3380)
Co-authored-by: Bruce Zhang <[email protected]>
1 parent 30eb72b commit c0f7814

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)