From 106ca8cb32302f27be463027f240eade3f639993 Mon Sep 17 00:00:00 2001 From: Rajendra Adhikari Date: Mon, 6 May 2024 11:37:26 -0500 Subject: [PATCH] Update main.py Small bugfix for finding matching files. --- buildstock_query/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/buildstock_query/main.py b/buildstock_query/main.py index b24c49b..91cfeae 100644 --- a/buildstock_query/main.py +++ b/buildstock_query/main.py @@ -322,7 +322,7 @@ def _download_results_csv(self) -> str: if 'Contents' not in s3_data: raise ValueError(f"Results parquet not found in s3 at {baseline_path}") matching_files = [path['Key'] for path in s3_data['Contents'] - if "up00.parquet" in path['Key'] or 'baseline' in path['Key']] + if "up00.parquet" in path['Key'] or 'baseline.parquet' in path['Key']] if len(matching_files) > 1: raise ValueError(f"Multiple results parquet found in s3 at {baseline_path} for baseline."