From 408dab58c4ed110df5948fde865fec8bec3f7119 Mon Sep 17 00:00:00 2001 From: tylertjburns <52676131+tylertjburns@users.noreply.github.com> Date: Thu, 5 Aug 2021 09:37:24 -0500 Subject: [PATCH] Update object_mapper.py type check validation more flexible to allow for inherited/abstract classes --- mapper/object_mapper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mapper/object_mapper.py b/mapper/object_mapper.py index 381a524..2ad874a 100644 --- a/mapper/object_mapper.py +++ b/mapper/object_mapper.py @@ -101,10 +101,10 @@ def create_map(self, type_from, type_to, mapping=None): :return: None """ - if (type(type_from) is not type): + if (not isinstance(type_from, type)): raise ObjectMapperException("type_from must be a type") - if (type(type_to) is not type): + if (not isinstance(type_to, type)): raise ObjectMapperException("type_to must be a type") if (mapping is not None and not isinstance(mapping, dict)):