Filter None in other util

benieric · benieric · commit b08d9d2e5023 · 2024-12-09T11:47:19.000-08:00
diff --git a/src/sagemaker/serve/model_format/mlflow/utils.py b/src/sagemaker/serve/model_format/mlflow/utils.py
@@ -350,7 +350,7 @@ def _select_container_for_mlflow_model(
     )
 
     image_uri = None
-    for casted_version in casted_versions:
+    for casted_version in filter(None, casted_versions):
         try:
             image_uri = image_uris.retrieve(
                 framework=framework_to_use,

Original file line number	Diff line number	Diff line change
`@@ -350,7 +350,7 @@ def _select_container_for_mlflow_model(`
`350`	`350`	`)`
`351`	`351`
`352`	`352`	`image_uri = None`
`353`		`- for casted_version in casted_versions:`
	`353`	`+ for casted_version in filter(None, casted_versions):`
`354`	`354`	`try:`
`355`	`355`	`image_uri = image_uris.retrieve(`
`356`	`356`	`framework=framework_to_use,`