diff --git a/pyxem/signals/polar_diffraction2d.py b/pyxem/signals/polar_diffraction2d.py index 88e7fa841..502581de0 100644 --- a/pyxem/signals/polar_diffraction2d.py +++ b/pyxem/signals/polar_diffraction2d.py @@ -274,7 +274,7 @@ def subtract_diffraction_background( For 'radial median' no extra parameters are necessary. - For 'radial percentile' the 'percentile' argument decides + For 'radial percentile' the 'percentile' argument decides which percentile to substract. **kwargs : To be passed to the chosen method. @@ -296,7 +296,13 @@ def subtract_diffraction_background( ) subtraction_function = method_dict[method] - return self.map(subtraction_function, inplace=inplace, **kwargs) + return self.map( + subtraction_function, + inplace=inplace, + output_dtype=self.data.dtype, + output_signal_size=self.axes_manager._signal_shape_in_array, + **kwargs, + ) class LazyPolarDiffraction2D(LazySignal, PolarDiffraction2D):