diff --git a/arpes/endstations/plugin/nexus.py b/arpes/endstations/plugin/nexus.py index aa91d0dc..478522cb 100644 --- a/arpes/endstations/plugin/nexus.py +++ b/arpes/endstations/plugin/nexus.py @@ -4,7 +4,7 @@ import h5py import numpy as np import xarray as xr -from pint import DimensionalityError, Quantity +from pint import Quantity from arpes.config import ureg from arpes.endstations import SingleFileEndstation, add_endstation @@ -30,10 +30,6 @@ } -class IncompatibleUnitsError(Exception): - pass - - class NeXusEndstation(SingleFileEndstation): """An endstation for reading arpes data from a nexus file.""" @@ -135,15 +131,9 @@ def load_nx_data(nxdata: h5py.Group, attributes: dict) -> xr.DataArray: coords = {} for axis, new_axis in zip(axes, new_axes): - if "units" not in nxdata[axis].attrs: - raise IncompatibleUnitsError(f"Axis {axis} does not have units.") coords[new_axis] = nxdata[axis][:] * ureg(nxdata[axis].attrs["units"]) - try: + if coords[new_axis].units == "degree": coords[new_axis] = coords[new_axis].to(ureg.rad) - except DimensionalityError as exc: - raise IncompatibleUnitsError( - f"Unit {coords[new_axis].units} of axis {axis} is not a unit of angle." - ) from exc data = nxdata[nxdata.attrs["signal"]][:] dims = new_axes