diff --git a/src/hope_country_report/web/views/charts.py b/src/hope_country_report/web/views/charts.py index 2cb7152c..fb7a40cc 100644 --- a/src/hope_country_report/web/views/charts.py +++ b/src/hope_country_report/web/views/charts.py @@ -2,6 +2,7 @@ from django.contrib.auth.mixins import PermissionRequiredMixin from django.http import HttpResponseBase +from django.db.models.query import QuerySet from django.views.generic import DetailView, ListView from hope_country_report.apps.power_query.utils import to_dataset @@ -15,7 +16,7 @@ class ChartListView(SelectedOfficeMixin, PermissionRequiredMixin, ListView[Chart permission_required = ["power_query.view_chartpage"] model = ChartPage - def get_queryset(self): + def get_queryset(self) -> QuerySet[ChartPage]: """ Filter the queryset to return only ChartPages related to the selected CountryOffice. """