hotfix - fix export report type

shapiromatron · Oct 7, 2023 · 0aac8d2 · 0aac8d2
1 parent ddebd4d
commit 0aac8d2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/hawc/apps/assessment/actions/audit.py b/hawc/apps/assessment/actions/audit.py
@@ -3,6 +3,7 @@
 import pandas as pd
 from django.contrib.contenttypes.models import ContentType
 from django.db.models import QuerySet
+from rest_framework.response import Response
 from reversion.models import Version
 
 from ...common.helper import FlatExport
@@ -246,7 +247,7 @@ def get_queryset(self):
         qs = getattr(self, method)()
         return qs.select_related("content_type", "revision")
 
-    def export(self) -> FlatExport:
+    def export(self) -> Response:
         qs = self.get_queryset()
         df = pd.DataFrame(
             qs.values_list(

diff --git a/hawc/apps/assessment/api/viewsets.py b/hawc/apps/assessment/api/viewsets.py
@@ -472,8 +472,7 @@ def add_item(app, count, title, url_route, modal_key):
     def logs(self, request: Request, pk: int, type: str):
         instance = self.get_object()
         serializer = AssessmentAuditSerializer.from_drf(data=dict(assessment=instance, type=type))
-        export = serializer.export()
-        return Response(export)
+        return serializer.export()
 
     @action(
         detail=False,