From ff2a6c273f7800daae0f552a3a12f85a6a209224 Mon Sep 17 00:00:00 2001 From: Ishmeet Bindra Date: Tue, 9 Jan 2024 00:51:20 +0530 Subject: [PATCH] Using get to fetch value --- pydanticrud/backends/dynamodb.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pydanticrud/backends/dynamodb.py b/pydanticrud/backends/dynamodb.py index bb26c1e..bff2818 100644 --- a/pydanticrud/backends/dynamodb.py +++ b/pydanticrud/backends/dynamodb.py @@ -141,7 +141,8 @@ def _get_type_possibilities(self, field_name) -> Set[tuple]: def type_from_definition(definition_signature: Union[str, dict]) -> dict: if isinstance(definition_signature, str): t = definition_signature.split("/")[-1] - return self.definitions.get(t) + if t in self.definitions: + return self.definitions.get(t) return definition_signature type_dicts = [type_from_definition(t) for t in possible_types]