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 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" ,
4242 "us-west-1" : " 710691900526" ,
4343 "us-west-2" : " 301217895009"
4444 },
45- "repository" : " sagemaker-neo -tensorflow"
45+ "repository" : " sagemaker-inference -tensorflow"
4646 }
4747 }
4848}
Original file line number Diff line number Diff line change @@ -63,7 +63,7 @@ def test_algo_uris(algo):
6363
6464def _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 :
You can’t perform that action at this time.
0 commit comments