Fix PR feeedback

viljarjf · Apr 21, 2024 · 58baa3a · 58baa3a
1 parent ae8be66
commit 58baa3a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git 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):