diff --git a/djorm_pgfulltext/models.py b/djorm_pgfulltext/models.py index a7a7a73..a995fa2 100644 --- a/djorm_pgfulltext/models.py +++ b/djorm_pgfulltext/models.py @@ -85,6 +85,7 @@ class SearchManagerMixIn(object): https://docs.djangoproject.com/en/1.4/howto/initial-data/#providing-initial-sql-data """ + _queryset_class = SearchQuerySet def __init__(self, fields=None, @@ -122,9 +123,6 @@ def update_search_field(self, search_field=None, fields=None, using=None, config super(SearchManagerMixIn, self).contribute_to_class(cls, name) - def get_queryset(self): - return SearchQuerySet(model=self.model, using=self._db) - def search(self, *args, **kwargs): return self.get_queryset().search(*args, **kwargs)