From 8328a4d5208fa68fbc480631582fbd26522f501d Mon Sep 17 00:00:00 2001 From: DinoBektesevic Date: Thu, 14 Mar 2024 12:52:42 -0700 Subject: [PATCH] Fix wrong file extensions check in IC. --- .../fits_standardizers/fits_standardizer.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/src/kbmod/standardizers/fits_standardizers/fits_standardizer.py b/src/kbmod/standardizers/fits_standardizers/fits_standardizer.py index a83c82f70..b80107781 100644 --- a/src/kbmod/standardizers/fits_standardizers/fits_standardizer.py +++ b/src/kbmod/standardizers/fits_standardizers/fits_standardizer.py @@ -102,17 +102,18 @@ def resolveFromPath(cls, tgt): ----- FileNotFoundError - when file doesn't exist """ + resources = {} canProcess = False # nasty hack, should do better extensions fname = Path(tgt) - extensions = fname.suffixes + extensions = "".join(fname.suffixes) # if the extensions are empty, we don't think it's a FITS file if not extensions: - return False, {} + return False, resources - if extensions[-1] in cls.valid_extensions: + if extensions in cls.valid_extensions: try: hdulist = fits.open(tgt) except OSError: @@ -120,9 +121,11 @@ def resolveFromPath(cls, tgt): # FileNotFoundError - bad file, let it raise pass else: + # otherwise it's a success! + resources = {"hdulist": hdulist} canProcess = True - return canProcess, {"hdulist": hdulist} + return canProcess, resources @classmethod def resolveTarget(cls, tgt):