added pandas-stubs and fixed linting

praekeltfoundation · Aug 28, 2024 · 2df6616 · 2df6616
1 parent 8a72a65
commit 2df6616
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 2 deletions.
diff --git a/home/export_ordered_sets.py b/home/export_ordered_sets.py
@@ -4,7 +4,7 @@
 from dataclasses import asdict, astuple, dataclass, fields
 from math import ceil
 
-from django.http import HttpResponse  # type: ignore 
+from django.http import HttpResponse  # type: ignore
 from openpyxl.styles import Font, NamedStyle
 from openpyxl.utils import get_column_letter
 from openpyxl.workbook import Workbook

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -78,6 +78,7 @@ types-openpyxl = "^3.1.0.20240301"
 types-requests = "^2.31.0.20240311"
 torch = "^2.4.0"
 pandas = "^2.2.2"
+pandas-stubs = "^2.2.2.240807"
 
 [[tool.poetry.source]]
 name = "pytorch-stable-cpu"