Skip to content

Commit 6dcafe4

Browse files
authored
Fix: image_detector skip None values in casted_versions (#4956)
* Fix: image_detector * Filter None in other util
1 parent bbb210a commit 6dcafe4

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
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,

src/sagemaker/serve/model_format/mlflow/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -350,7 +350,7 @@ def _select_container_for_mlflow_model(
350350
)
351351

352352
image_uri = None
353-
for casted_version in casted_versions:
353+
for casted_version in filter(None, casted_versions):
354354
try:
355355
image_uri = image_uris.retrieve(
356356
framework=framework_to_use,

0 commit comments

Comments
 (0)