From 8ae729be4dd90c028b4d6edf80a6f2a2a869c03b Mon Sep 17 00:00:00 2001 From: Dan Knight <56950335+dan-knight@users.noreply.github.com> Date: Mon, 5 Feb 2024 01:02:37 -0800 Subject: [PATCH 1/2] Fix closed file bug in CSV loader --- hiddenfb/utility/file_loader/csv.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/hiddenfb/utility/file_loader/csv.py b/hiddenfb/utility/file_loader/csv.py index 59545d0..6c6b8dd 100644 --- a/hiddenfb/utility/file_loader/csv.py +++ b/hiddenfb/utility/file_loader/csv.py @@ -12,9 +12,9 @@ def load(self, path: Path) -> List[Dict[str, Any]]: with self._load_file(path, file_type="csv") as f: data = csv.reader(f) - header_columns: List[str] = self._parse_header(data) - - return flatten([self._parse_row(row, header=header_columns) for row in data]) + header_columns: List[str] = self._parse_header(data) + + return flatten([self._parse_row(row, header=header_columns) for row in data]) def _parse_header(self, reader: Iterator[Any]) -> List[str]: return next(reader) From a091f09f462d7bc7e511d5b273a3d4f222b68a28 Mon Sep 17 00:00:00 2001 From: dan-knight Date: Mon, 5 Feb 2024 01:05:30 -0800 Subject: [PATCH 2/2] Fix code style --- hiddenfb/utility/file_loader/csv.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/hiddenfb/utility/file_loader/csv.py b/hiddenfb/utility/file_loader/csv.py index 6c6b8dd..6d7f05f 100644 --- a/hiddenfb/utility/file_loader/csv.py +++ b/hiddenfb/utility/file_loader/csv.py @@ -13,8 +13,10 @@ def load(self, path: Path) -> List[Dict[str, Any]]: data = csv.reader(f) header_columns: List[str] = self._parse_header(data) - - return flatten([self._parse_row(row, header=header_columns) for row in data]) + + return flatten( + [self._parse_row(row, header=header_columns) for row in data] + ) def _parse_header(self, reader: Iterator[Any]) -> List[str]: return next(reader)