diff --git a/mongoengine/queryset/queryset.py b/mongoengine/queryset/queryset.py index 0b065853f..caa6bc43d 100644 --- a/mongoengine/queryset/queryset.py +++ b/mongoengine/queryset/queryset.py @@ -806,6 +806,7 @@ def hint(self, index=None): queryset = self.clone() queryset._hint = index + # If a cursor object has already been created, apply the hint to it. if queryset._cursor_obj: queryset._cursor_obj.hint(queryset._hint) @@ -813,8 +814,12 @@ def hint(self, index=None): def batch_size(self, size): queryset = self.clone() - queryset._cursor.batch_size(size) queryset._batch_size = size + + # If a cursor object has already been created, apply the batch size to it. + if queryset._cursor_obj: + queryset._cursor_obj.batch_size(queryset._batch_size) + return queryset def distinct(self, field, dereference=True):