From 8cc50d9aa1d053830ad187441cf2b46ff1780e57 Mon Sep 17 00:00:00 2001 From: Eero Vaher Date: Wed, 4 Sep 2024 15:31:00 +0200 Subject: [PATCH] Simplify converting units to strings (#508) --- gwcs/wcs.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/gwcs/wcs.py b/gwcs/wcs.py index 62173532..6dede33e 100644 --- a/gwcs/wcs.py +++ b/gwcs/wcs.py @@ -6,7 +6,6 @@ import astropy.io.fits as fits import numpy as np import numpy.linalg as npla -from astropy import units as u from astropy.modeling import fix_inputs, projections from astropy.modeling.bounding_box import CompoundBoundingBox from astropy.modeling.bounding_box import ModelBoundingBox as Bbox @@ -2031,16 +2030,12 @@ def find_frame(axis_number): # index of the axis in this frame's fidx = frame.axes_order.index(axno) - if hasattr(frame.unit[fidx], 'get_format_name'): - cunit = frame.unit[fidx].get_format_name(u.format.Fits).upper() - else: - cunit = '' axis_info = _WorldAxisInfo( axis=axno, frame=frame, world_axis_order=self.output_frame.axes_order.index(axno), - cunit=cunit, + cunit=frame.unit[fidx].to_string('fits', fraction=True).upper(), ctype=cf.get_ctype_from_ucd(self.world_axis_physical_types[axno]), input_axes=mapping[axno] )