diff --git a/src/saml2/__init__.py b/src/saml2/__init__.py index 93c4521..38631e4 100644 --- a/src/saml2/__init__.py +++ b/src/saml2/__init__.py @@ -844,6 +844,14 @@ def extension_elements_to_elements(extension_elements, schemas): according to the schemas. """ res = [] + + if isinstance(schemas, list): + pass + elif isinstance(schemas, dict): + schemas = schemas.values() + else: + return res + for extension_element in extension_elements: for schema in schemas: inst = extension_element_to_element(extension_element,