replace iterable class in qs.non_polymorphic

2018-11-28 17:23:00 +03:00 · 2018-11-28 17:23:00 +03:00 · c565ae101a
parent 6d27bb3334
commit c565ae101a
1 changed files with 2 additions and 0 deletions
--- 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):