From ac27aaea71cb9a035446f50d3dc0148ec144ec88 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Tue, 11 Feb 2025 23:14:30 +0000 Subject: add type checking to csv_reader function --- src/csv_reader.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'src/csv_reader.py') diff --git a/src/csv_reader.py b/src/csv_reader.py index 81f1485..23afc66 100644 --- a/src/csv_reader.py +++ b/src/csv_reader.py @@ -1,7 +1,8 @@ import csv from io import StringIO +from typing import List, Dict -def csv_reader(file): - f = StringIO(file) +def csv_reader(content: str) -> List[Dict[str, str]]: + f = StringIO(content) reader = csv.DictReader(f) return list(reader) -- cgit v1.2.3