Ignore BinTableHDU's and HDU's lacking shape

This commit is contained in:
Henrik Norman 2021-09-29 18:43:43 +02:00
Родитель 3d23dd7ec7
Коммит a64a727703
1 изменённых файлов: 4 добавлений и 1 удалений

Просмотреть файл

@ -71,9 +71,12 @@ class SimpleFitsCollection(ImageCollection):
hdu = hdul[self._hdu_index] hdu = hdul[self._hdu_index]
else: else:
for hdu in hdul: for hdu in hdul:
if len(hdu.shape) > 1: if (hasattr(hdu, 'shape') and len(hdu.shape) > 1
and type(hdu) is not fits.hdu.table.BinTableHDU):
break break
if type(hdu) is fits.hdu.table.BinTableHDU:
raise Exception(f'cannot process input `{fits_path}`: Did not find any HDU with image data')
wcs = WCS(hdu.header) wcs = WCS(hdu.header)
shape = hdu.shape shape = hdu.shape