From 40150eaa90b54174544a85c17b6adb585e9177fd Mon Sep 17 00:00:00 2001 From: Isabelle-C Date: Sat, 7 Oct 2023 00:40:46 -0500 Subject: [PATCH] Attempt to fix mypy error --- .../output/parse_growth_file.py | 21 ++++++++++--------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/src/arcade_collection/output/parse_growth_file.py b/src/arcade_collection/output/parse_growth_file.py index 7f674a4..581459e 100644 --- a/src/arcade_collection/output/parse_growth_file.py +++ b/src/arcade_collection/output/parse_growth_file.py @@ -48,16 +48,17 @@ def parse_growth_file(tar: tarfile.TarFile) -> pd.DataFrame: all_timepoints = [] for member in tar.getmembers(): extracted_member = tar.extractfile(member) - extracted_json = json.loads(extracted_member.read().decode("utf-8")) - seed = extracted_json["seed"] - - all_timepoints.extend( - [ - data - for timepoint in extracted_json["timepoints"] - for data in parse_growth_timepoint(timepoint, seed) - ] - ) + if extracted_member is not None: + extracted_json = json.loads(extracted_member.read().decode("utf-8")) + seed = extracted_json["seed"] + + all_timepoints.extend( + [ + data + for timepoint in extracted_json["timepoints"] + for data in parse_growth_timepoint(timepoint, seed) + ] + ) timepoints_df = pd.DataFrame(all_timepoints, columns=GROWTH_COLUMNS) return timepoints_df