diff --git a/arpes/endstations/plugin/ANTARES.py b/arpes/endstations/plugin/ANTARES.py index 4a34792d..290c54f5 100644 --- a/arpes/endstations/plugin/ANTARES.py +++ b/arpes/endstations/plugin/ANTARES.py @@ -7,7 +7,6 @@ import h5py import numpy as np -from numpy._typing import NDArray import xarray as xr from arpes.endstations import ( @@ -25,6 +24,7 @@ if TYPE_CHECKING: from _typeshed import Incomplete + from numpy.typing import NDArray __all__ = ("ANTARESEndstation",) @@ -115,7 +115,20 @@ def load_top_level_scan( scan_desc: SCANDESC | None = None, spectrum_index=None, ) -> xr.Dataset: - """Reads a spectrum from the top level group in a NeXuS scan format.""" + """Reads a spectrum from the top level group in a NeXuS scan format. + + [TODO:description] + + Args: + group ([TODO:type]): [TODO:description] + scan_desc: [TODO:description] + spectrum_index ([TODO:type]): [TODO:description] + + Returns: + [TODO:description] + """ + if scan_desc: + warnings.warn("scan_desc is not supported", stacklevel=2) dr = self.read_scan_data(group) bindings = read_data_attributes_from_tree(group, READ_TREE) @@ -181,7 +194,9 @@ def take_last(vs): return vs - for dim_order, name, values in zip(actuator_dim_order, actuator_names, actuator_list): + for dim_order, name, values in zip( + actuator_dim_order, actuator_names, actuator_list, strict=False + ): name = self.RENAME_KEYS.get(name, name) dims[dim_order] = name coords[name] = take_last(values) diff --git a/arpes/xarray_extensions.py b/arpes/xarray_extensions.py index ea19afec..d28ce226 100644 --- a/arpes/xarray_extensions.py +++ b/arpes/xarray_extensions.py @@ -438,7 +438,6 @@ def transpose_to_front(self, dim: str) -> xr.DataArray | xr.Dataset: def transpose_to_back(self, dim: str) -> xr.DataArray | xr.Dataset: """Transpose the dimensions (to back). - [TODO:description] Args: dim: dimension to back