Skip to content

Commit 7d1916d

Browse files
kkothari22Kothariicywang86rui
authored
feature: update image uri for neo tensorflow (#1967)
Co-authored-by: Kothari <[email protected]> Co-authored-by: icywang86rui <[email protected]>
1 parent 99b70a1 commit 7d1916d

File tree

2 files changed

+14
-14
lines changed

2 files changed

+14
-14
lines changed

src/sagemaker/image_uri_config/neo-tensorflow.json

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -2,20 +2,20 @@
22
"processors": ["cpu", "gpu"],
33
"scope": ["inference"],
44
"version_aliases": {
5-
"1.4.1": "1.15.0",
6-
"1.5.0": "1.15.0",
7-
"1.6.0": "1.15.0",
8-
"1.7.0": "1.15.0",
9-
"1.8.0": "1.15.0",
10-
"1.9.0": "1.15.0",
11-
"1.10.0": "1.15.0",
12-
"1.11.0": "1.15.0",
13-
"1.12.0": "1.15.0",
14-
"1.13.0": "1.15.0",
15-
"1.14.0": "1.15.0"
5+
"1.4.1": "1.15.3",
6+
"1.5.0": "1.15.3",
7+
"1.6.0": "1.15.3",
8+
"1.7.0": "1.15.3",
9+
"1.8.0": "1.15.3",
10+
"1.9.0": "1.15.3",
11+
"1.10.0": "1.15.3",
12+
"1.11.0": "1.15.3",
13+
"1.12.0": "1.15.3",
14+
"1.13.0": "1.15.3",
15+
"1.14.0": "1.15.3"
1616
},
1717
"versions": {
18-
"1.15.0": {
18+
"1.15.3": {
1919
"py_versions": ["py3"],
2020
"registries": {
2121
"af-south-1": "774647643957",
@@ -42,7 +42,7 @@
4242
"us-west-1": "710691900526",
4343
"us-west-2": "301217895009"
4444
},
45-
"repository": "sagemaker-neo-tensorflow"
45+
"repository": "sagemaker-inference-tensorflow"
4646
}
4747
}
4848
}

tests/unit/sagemaker/image_uris/test_neo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def test_algo_uris(algo):
6363

6464
def _test_neo_framework_uris(framework, version):
6565
framework_in_config = f"neo-{framework}"
66-
framework_in_uri = f"neo-{framework}" if framework == "tensorflow" else f"inference-{framework}"
66+
framework_in_uri = f"inference-{framework}"
6767

6868
for region in regions.regions():
6969
if region in ACCOUNTS:

0 commit comments

Comments
 (0)