Merge pull request #371 from tumb1er/replace_iterable_class

replace iterable class in qs.non_polymorphic
2019-07-11 22:03:07 +02:00 · 2019-07-11 22:03:07 +02:00 · f9492a12a8
parent f06a02a3d0 c565ae101a
commit f9492a12a8
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):