Skip to content

Commit 5b70162

Browse files
committed
Fix: image_detector
1 parent 8d08c99 commit 5b70162

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/sagemaker/image_uri_config/xgboost.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
{
22
"inference": {
33
"version_aliases": {
4-
"latest": "1.7-1"
4+
"latest": "1"
55
},
66
"versions": {
77
"1": {

src/sagemaker/serve/detector/image_detector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def auto_detect_container(model, region: str, instance_type: str) -> str:
4343
casted_versions = _cast_to_compatible_version(fw, fw_version) if fw_version else (None,)
4444
dlc = None
4545

46-
for casted_version in casted_versions:
46+
for casted_version in filter(None, casted_versions):
4747
try:
4848
dlc = image_uris.retrieve(
4949
framework=fw,

0 commit comments

Comments
 (0)