style(cli): docstring tweaking.

paquiteau · Feb 24, 2023 · 28ea018 · 28ea018
1 parent 58486c1
commit 28ea018
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/patch_denoise/bindings/cli.py b/src/patch_denoise/bindings/cli.py
@@ -120,7 +120,7 @@ def main():
 
     input_data, affine = load_as_array(args.input_file)
     if args.mask == "auto":
-        mask = compute_mask(input_data, time_axis=-1)
+        mask = compute_mask(input_data)
         affine_mask = None
     else:
         mask, affine_mask = load_as_array(args.mask)

diff --git a/src/patch_denoise/bindings/utils.py b/src/patch_denoise/bindings/utils.py
@@ -143,11 +143,11 @@ def compute_mask(array, convex=False):
         Array to compute mask for.
     convex : bool, default False
         If True, the mask is convex for each slice.
+
     Returns
     -------
     numpy.ndarray
         Mask for array.
-
     """
     if time_axis is not None:
         mean = array.mean(axis=-1)