File tree Expand file tree Collapse file tree 2 files changed +14
-14
lines changed
src/sagemaker/image_uri_config
tests/unit/sagemaker/image_uris Expand file tree Collapse file tree 2 files changed +14
-14
lines changed Original file line number Diff line number Diff line change 2
2
"processors" : [" cpu" , " gpu" ],
3
3
"scope" : [" inference" ],
4
4
"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 "
16
16
},
17
17
"versions" : {
18
- "1.15.0 " : {
18
+ "1.15.3 " : {
19
19
"py_versions" : [" py3" ],
20
20
"registries" : {
21
21
"af-south-1" : " 774647643957" ,
42
42
"us-west-1" : " 710691900526" ,
43
43
"us-west-2" : " 301217895009"
44
44
},
45
- "repository" : " sagemaker-neo -tensorflow"
45
+ "repository" : " sagemaker-inference -tensorflow"
46
46
}
47
47
}
48
48
}
Original file line number Diff line number Diff line change @@ -63,7 +63,7 @@ def test_algo_uris(algo):
63
63
64
64
def _test_neo_framework_uris (framework , version ):
65
65
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 } "
67
67
68
68
for region in regions .regions ():
69
69
if region in ACCOUNTS :
You can’t perform that action at this time.
0 commit comments