Checking if the key is present

Author: sreedes

src/sagemaker/session.py

@@ -2521,7 +2521,7 @@ def _create_model_request(
         request = {"ModelName": name, "ExecutionRoleArn": role}
         if isinstance(container_definition, list):
             request["Containers"] = container_definition
-        elif container_definition["ModelPackageName"] is not None:
+        elif 'ModelPackageName' in container_definition:
             request["Containers"] = [container_definition]
         else:
             request["PrimaryContainer"] = container_definition