From c565ae101afd63226280e6cad26c9df95a1f9bf0 Mon Sep 17 00:00:00 2001 From: tumb1er Date: Wed, 28 Nov 2018 17:23:00 +0300 Subject: [PATCH] replace iterable class in qs.non_polymorphic --- polymorphic/query.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/polymorphic/query.py b/polymorphic/query.py index 3d9d28c..50f4953 100644 --- a/polymorphic/query.py +++ b/polymorphic/query.py @@ -138,6 +138,8 @@ class PolymorphicQuerySet(QuerySet): base class used for this query are returned.""" qs = self._clone() qs.polymorphic_disabled = True + if issubclass(qs._iterable_class, PolymorphicModelIterable): + qs._iterable_class = ModelIterable return qs def instance_of(self, *args):