diff --git a/apps/favorites/views.py b/apps/favorites/views.py index 5d99ed4b..6b7bdab0 100644 --- a/apps/favorites/views.py +++ b/apps/favorites/views.py @@ -1,6 +1,7 @@ """Views for app favorites.""" from rest_framework import generics from establishment.models import Establishment +from establishment.filters import EstablishmentFilter from establishment.serializers import EstablishmentBaseSerializer from .models import Favorites @@ -17,6 +18,7 @@ class FavoritesEstablishmentListView(generics.ListAPIView): """List views for favorites""" serializer_class = EstablishmentBaseSerializer + filter_class = EstablishmentFilter def get_queryset(self): """Override get_queryset method"""