diff --git a/romancal/assign_wcs/assign_wcs_step.py b/romancal/assign_wcs/assign_wcs_step.py index c9efd3242..f61c19b85 100644 --- a/romancal/assign_wcs/assign_wcs_step.py +++ b/romancal/assign_wcs/assign_wcs_step.py @@ -93,8 +93,12 @@ def load_wcs(input_model, reference_files=None): axes_names=("v2", "v3"), unit=(u.arcsec, u.arcsec), ) - v2v3vacorr = cf.Frame2D(name='v2v3vacorr', axes_order=(0, 1), - axes_names=('v2', 'v3'), unit=(u.arcsec, u.arcsec)) + v2v3vacorr = cf.Frame2D( + name="v2v3vacorr", + axes_order=(0, 1), + axes_names=("v2", "v3"), + unit=(u.arcsec, u.arcsec), + ) world = cf.CelestialFrame(reference_frame=coord.ICRS(), name="world") # Transforms between frames @@ -105,7 +109,7 @@ def load_wcs(input_model, reference_files=None): va_corr = pointing.dva_corr_model( va_scale=input_model.meta.velocity_aberration.scale_factor, v2_ref=input_model.meta.wcsinfo.v2_ref, - v3_ref=input_model.meta.wcsinfo.v3_ref + v3_ref=input_model.meta.wcsinfo.v3_ref, ) pipeline = [ diff --git a/romancal/assign_wcs/pointing.py b/romancal/assign_wcs/pointing.py index edd967d06..b9b351c7a 100644 --- a/romancal/assign_wcs/pointing.py +++ b/romancal/assign_wcs/pointing.py @@ -80,7 +80,9 @@ def dva_corr_model(va_scale, v2_ref, v3_ref): if va_scale <= 0: raise ValueError("'Velocity aberration scale must be a positive number.") - va_corr = Scale(va_scale, name='dva_scale_v2') & Scale(va_scale, name='dva_scale_v3') + va_corr = Scale(va_scale, name="dva_scale_v2") & Scale( + va_scale, name="dva_scale_v3" + ) if v2_ref is None: v2_ref = 0 @@ -97,6 +99,8 @@ def dva_corr_model(va_scale, v2_ref, v3_ref): v2_shift = (1 - va_scale) * v2_ref v3_shift = (1 - va_scale) * v3_ref - va_corr |= Shift(v2_shift, name='dva_v2_shift') & Shift(v3_shift, name='dva_v3_shift') - va_corr.name = 'DVA_Correction' + va_corr |= Shift(v2_shift, name="dva_v2_shift") & Shift( + v3_shift, name="dva_v3_shift" + ) + va_corr.name = "DVA_Correction" return va_corr