From 857b5ae39e0d45387ab1c562964293965f1b6e56 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 14 Aug 2024 18:25:36 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- RMutils/util_misc.py | 26 +++++++++++++++++++++----- 1 file changed, 21 insertions(+), 5 deletions(-) diff --git a/RMutils/util_misc.py b/RMutils/util_misc.py index 13957e6..199b782 100644 --- a/RMutils/util_misc.py +++ b/RMutils/util_misc.py @@ -93,23 +93,39 @@ def update_position_wcsaxes(header): # Store and delete the WCSAXES keyword - wcsaxes = header['WCSAXES'] - del header['WCSAXES'] + wcsaxes = header["WCSAXES"] + del header["WCSAXES"] # Determine the correct insertion point before the first WCS-related keyword - wcs_keywords = ['CRPIX1', 'CRPIX2', 'CRVAL1', 'CRVAL2', 'CTYPE1', 'CTYPE2', 'CUNIT1', 'CUNIT2', 'PC1_1', 'PC2_2', 'CD1_1', 'CD2_2'] + wcs_keywords = [ + "CRPIX1", + "CRPIX2", + "CRVAL1", + "CRVAL2", + "CTYPE1", + "CTYPE2", + "CUNIT1", + "CUNIT2", + "PC1_1", + "PC2_2", + "CD1_1", + "CD2_2", + ] # Convert the header keys to a list header_keys = list(header.keys()) # Find the first occurrence of any WCS-related keyword - insert_pos = min(header_keys.index(key) for key in wcs_keywords if key in header_keys) + insert_pos = min( + header_keys.index(key) for key in wcs_keywords if key in header_keys + ) # Insert WCSAXES at the correct position - header.insert(insert_pos, ('WCSAXES', wcsaxes)) + header.insert(insert_pos, ("WCSAXES", wcsaxes)) return header + def remove_header_third_fourth_axis(header): """Removes extra axes from header to compress down to 2 axes""" # List of keys related to the 3rd and 4th axes to remove (essentially everything with a '3' or '4')