Skip to content

Commit 993e716

Browse files
Dan ChoiChoiByungWook
Dan Choi
authored and
ChoiByungWook
committed
feature: add tensorflow 1.15.4 and 2.3.1 as valid versions
1 parent 5a0530f commit 993e716

File tree

3 files changed

+135
-6
lines changed

3 files changed

+135
-6
lines changed

‎src/sagemaker/image_uri_config/tensorflow.json

Lines changed: 128 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,7 @@
232232
"1.12": "1.12.0",
233233
"1.13": "1.13.0",
234234
"1.14": "1.14.0",
235-
"1.15": "1.15.3",
235+
"1.15": "1.15.4",
236236
"1.4": "1.4.1",
237237
"1.5": "1.5.0",
238238
"1.6": "1.6.0",
@@ -242,7 +242,7 @@
242242
"2.0": "2.0.1",
243243
"2.1": "2.1.1",
244244
"2.2": "2.2.0",
245-
"2.3": "2.3.0"
245+
"2.3": "2.3.1"
246246
},
247247
"versions": {
248248
"1.10.0": {
@@ -480,6 +480,35 @@
480480
},
481481
"repository": "tensorflow-inference"
482482
},
483+
"1.15.4": {
484+
"registries": {
485+
"af-south-1": "626614931356",
486+
"ap-east-1": "871362719292",
487+
"ap-northeast-1": "763104351884",
488+
"ap-northeast-2": "763104351884",
489+
"ap-south-1": "763104351884",
490+
"ap-southeast-1": "763104351884",
491+
"ap-southeast-2": "763104351884",
492+
"ca-central-1": "763104351884",
493+
"cn-north-1": "727897471807",
494+
"cn-northwest-1": "727897471807",
495+
"eu-central-1": "763104351884",
496+
"eu-north-1": "763104351884",
497+
"eu-south-1": "692866216735",
498+
"eu-west-1": "763104351884",
499+
"eu-west-2": "763104351884",
500+
"eu-west-3": "763104351884",
501+
"me-south-1": "217643126080",
502+
"sa-east-1": "763104351884",
503+
"us-east-1": "763104351884",
504+
"us-east-2": "763104351884",
505+
"us-gov-west-1": "442386744353",
506+
"us-iso-east-1": "886529160074",
507+
"us-west-1": "763104351884",
508+
"us-west-2": "763104351884"
509+
},
510+
"repository": "tensorflow-inference"
511+
},
483512
"1.4.1": {
484513
"py_versions": [
485514
"py2"
@@ -845,6 +874,35 @@
845874
"us-west-2": "763104351884"
846875
},
847876
"repository": "tensorflow-inference"
877+
},
878+
"2.3.1": {
879+
"registries": {
880+
"af-south-1": "626614931356",
881+
"ap-east-1": "871362719292",
882+
"ap-northeast-1": "763104351884",
883+
"ap-northeast-2": "763104351884",
884+
"ap-south-1": "763104351884",
885+
"ap-southeast-1": "763104351884",
886+
"ap-southeast-2": "763104351884",
887+
"ca-central-1": "763104351884",
888+
"cn-north-1": "727897471807",
889+
"cn-northwest-1": "727897471807",
890+
"eu-central-1": "763104351884",
891+
"eu-north-1": "763104351884",
892+
"eu-south-1": "692866216735",
893+
"eu-west-1": "763104351884",
894+
"eu-west-2": "763104351884",
895+
"eu-west-3": "763104351884",
896+
"me-south-1": "217643126080",
897+
"sa-east-1": "763104351884",
898+
"us-east-1": "763104351884",
899+
"us-east-2": "763104351884",
900+
"us-gov-west-1": "442386744353",
901+
"us-iso-east-1": "886529160074",
902+
"us-west-1": "763104351884",
903+
"us-west-2": "763104351884"
904+
},
905+
"repository": "tensorflow-inference"
848906
}
849907
}
850908
},
@@ -859,7 +917,7 @@
859917
"1.12": "1.12.0",
860918
"1.13": "1.13.1",
861919
"1.14": "1.14.0",
862-
"1.15": "1.15.3",
920+
"1.15": "1.15.4",
863921
"1.4": "1.4.1",
864922
"1.5": "1.5.0",
865923
"1.6": "1.6.0",
@@ -869,7 +927,7 @@
869927
"2.0": "2.0.1",
870928
"2.1": "2.1.1",
871929
"2.2": "2.2.0",
872-
"2.3": "2.3.0"
930+
"2.3": "2.3.1"
873931
},
874932
"versions": {
875933
"1.10.0": {
@@ -1164,6 +1222,40 @@
11641222
},
11651223
"repository": "tensorflow-training"
11661224
},
1225+
"1.15.4": {
1226+
"py_versions": [
1227+
"py3",
1228+
"py36",
1229+
"py37"
1230+
],
1231+
"registries": {
1232+
"af-south-1": "626614931356",
1233+
"ap-east-1": "871362719292",
1234+
"ap-northeast-1": "763104351884",
1235+
"ap-northeast-2": "763104351884",
1236+
"ap-south-1": "763104351884",
1237+
"ap-southeast-1": "763104351884",
1238+
"ap-southeast-2": "763104351884",
1239+
"ca-central-1": "763104351884",
1240+
"cn-north-1": "727897471807",
1241+
"cn-northwest-1": "727897471807",
1242+
"eu-central-1": "763104351884",
1243+
"eu-north-1": "763104351884",
1244+
"eu-south-1": "692866216735",
1245+
"eu-west-1": "763104351884",
1246+
"eu-west-2": "763104351884",
1247+
"eu-west-3": "763104351884",
1248+
"me-south-1": "217643126080",
1249+
"sa-east-1": "763104351884",
1250+
"us-east-1": "763104351884",
1251+
"us-east-2": "763104351884",
1252+
"us-gov-west-1": "442386744353",
1253+
"us-iso-east-1": "886529160074",
1254+
"us-west-1": "763104351884",
1255+
"us-west-2": "763104351884"
1256+
},
1257+
"repository": "tensorflow-training"
1258+
},
11671259
"1.4.1": {
11681260
"py_versions": [
11691261
"py2"
@@ -1551,6 +1643,38 @@
15511643
"us-west-2": "763104351884"
15521644
},
15531645
"repository": "tensorflow-training"
1646+
},
1647+
"2.3.1": {
1648+
"py_versions": [
1649+
"py37"
1650+
],
1651+
"registries": {
1652+
"af-south-1": "626614931356",
1653+
"ap-east-1": "871362719292",
1654+
"ap-northeast-1": "763104351884",
1655+
"ap-northeast-2": "763104351884",
1656+
"ap-south-1": "763104351884",
1657+
"ap-southeast-1": "763104351884",
1658+
"ap-southeast-2": "763104351884",
1659+
"ca-central-1": "763104351884",
1660+
"cn-north-1": "727897471807",
1661+
"cn-northwest-1": "727897471807",
1662+
"eu-central-1": "763104351884",
1663+
"eu-north-1": "763104351884",
1664+
"eu-south-1": "692866216735",
1665+
"eu-west-1": "763104351884",
1666+
"eu-west-2": "763104351884",
1667+
"eu-west-3": "763104351884",
1668+
"me-south-1": "217643126080",
1669+
"sa-east-1": "763104351884",
1670+
"us-east-1": "763104351884",
1671+
"us-east-2": "763104351884",
1672+
"us-gov-west-1": "442386744353",
1673+
"us-iso-east-1": "886529160074",
1674+
"us-west-1": "763104351884",
1675+
"us-west-2": "763104351884"
1676+
},
1677+
"repository": "tensorflow-training"
15541678
}
15551679
}
15561680
}

‎tests/conftest.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -205,11 +205,16 @@ def tensorflow_training_py_version(tensorflow_training_version, request):
205205

206206
@pytest.fixture(scope="module", params=["py2", "py3"])
207207
def tensorflow_inference_py_version(tensorflow_inference_version, request):
208+
version = Version(tensorflow_inference_version)
209+
if version == Version("1.15.4"):
210+
return "py36"
208211
return _tf_py_version(tensorflow_inference_version, request)
209212

210213

211214
def _tf_py_version(tf_version, request):
212215
version = Version(tf_version)
216+
if Version("1.15") <= version <= Version("1.15.4"):
217+
return "py3"
213218
if version < Version("1.11"):
214219
return "py2"
215220
if version < Version("2.2"):

‎tests/unit/sagemaker/image_uris/test_dlc_frameworks.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,11 +130,11 @@ def _expected_tf_training_uri(tf_training_version, py_version, processor="cpu",
130130
)
131131

132132

133-
def test_tensorflow_inference(tensorflow_inference_version):
133+
def test_tensorflow_inference(tensorflow_inference_version, tensorflow_inference_py_version):
134134
_test_image_uris(
135135
"tensorflow",
136136
tensorflow_inference_version,
137-
"py2",
137+
tensorflow_inference_py_version,
138138
"inference",
139139
_expected_tf_inference_uri,
140140
{"tf_inference_version": tensorflow_inference_version},

0 commit comments

Comments
 (0)