diff --git a/pynamodb_encoder/decoder.py b/pynamodb_encoder/decoder.py index 344b419..83601fc 100644 --- a/pynamodb_encoder/decoder.py +++ b/pynamodb_encoder/decoder.py @@ -43,7 +43,7 @@ def coerce(self, element_type: Union[Type[Attribute], None]) -> Attribute: return (element_type or Attribute)() def decode_map(self, attr: MapAttribute, data: Dict[str, Any]): - if type(attr) == MapAttribute: + if type(attr) is MapAttribute: return data elif isinstance(attr, DynamicMapAttribute): return self.decode_dynamic_map(attr, data) diff --git a/pynamodb_encoder/encoder.py b/pynamodb_encoder/encoder.py index 58dbed9..cf9c40e 100644 --- a/pynamodb_encoder/encoder.py +++ b/pynamodb_encoder/encoder.py @@ -37,7 +37,7 @@ def encode_list(self, attr: ListAttribute, data: List) -> List: return [self.encode_attribute(element_attr, value) for value in data] def encode_map(self, attr: MapAttribute, data: MapAttribute) -> Dict: - if type(attr) == MapAttribute: + if type(attr) is MapAttribute: return {name: data[name] for name in data} elif isinstance(attr, DynamicMapAttribute): return self.encode_dynamic_map(attr, data)