From 7f7d34313ce6def68831a43ca2f65e96386e6d21 Mon Sep 17 00:00:00 2001 From: Robert Stein Date: Thu, 28 Jul 2022 19:21:28 -0700 Subject: [PATCH] Enforce typing on ImageSelector --- winterdrp/processors/utils/image_selector.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/winterdrp/processors/utils/image_selector.py b/winterdrp/processors/utils/image_selector.py index 3446a1e11..94f93fae3 100644 --- a/winterdrp/processors/utils/image_selector.py +++ b/winterdrp/processors/utils/image_selector.py @@ -13,14 +13,17 @@ def select_from_images( target_values: str | list[str] = "science", ) -> tuple[list[np.ndarray], list[astropy.io.fits.Header]]: - if isinstance(target_values, str): - target_values = [target_values] + # Enforce string in list for later matching + if not isinstance(target_values, list): + target_values = [str(target_values)] + else: + target_values = [str(x) for x in target_values] passing_images = [] passing_headers = [] for i, header in enumerate(headers): - if header[header_key] in target_values: + if str(header[header_key]) in target_values: passing_images.append(images[i]) passing_headers.append(header)