fix: phase of complex methods returns np.ndarray

xarray-contrib · Jun 30, 2023 · dfb050d · dfb050d
1 parent 5921eb4
commit dfb050d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xeofs/models/rotator.py b/xeofs/models/rotator.py
@@ -93,7 +93,7 @@ def components_phase(self) -> DataArray | Dataset | DataArrayList:
  Phase of the components.
 
  '''
- comps = np.angle(self._components)
+ comps = xr.apply_ufunc(np.angle, self._components)
  comps.name = 'phase'
  comps = self._model.stacker.inverse_transform_components(comps)
  return comps