# -*- coding: utf-8 -*-
from rest_framework import mixins, viewsets

from .models import Notification
from .serializers import NotificationSerializer


class BaseNotificationViewSet(
    mixins.RetrieveModelMixin,
    mixins.ListModelMixin,
    viewsets.GenericViewSet
):
    serializer_class = NotificationSerializer
    model = Notification

    def get_queryset(self):
        return Notification.objects.live()
    
# -*- coding: utf-8 -*-
from rest_framework import mixins, viewsets

from .models import Notification
from .serializers import NotificationSerializer


class BaseNotificationViewSet(
    mixins.RetrieveModelMixin,
    mixins.ListModelMixin,
    viewsets.GenericViewSet
):
    serializer_class = NotificationSerializer
    model = Notification

    def get_queryset(self):
        return Notification.objects.live()

remove whitespace from end of file