fix(genai_core): handling bad FMs (improved finetuned)

aws-samples · Dec 4, 2023 · ac40a86 · ac40a86
1 parent f5e4440
commit ac40a86
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/shared/layers/python-sdk/python/genai_core/models.py b/lib/shared/layers/python-sdk/python/genai_core/models.py
@@ -103,8 +103,8 @@ def list_bedrock_finetuned_models():
             }
             for model in bedrock_custom_models
             # Exclude embeddings and stable diffusion models
-            if model.get("inputModalities", None) != None
-            and model.get("outputModalities", None) != None
+            if "inputModalities" in model
+            and "outputModalities" in model
             and Modality.EMBEDDING.value not in model.get("outputModalities", [])
             and Modality.IMAGE.value not in model.get("outputModalities", [])
         ]