Skip to content

Commit 0f2df10

Browse files
committed
docstring update and linting
1 parent 6531dab commit 0f2df10

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

httomolibgpu/prep/stripe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -644,7 +644,7 @@ def remove_stripe_fw(
644644
level : int, optional
645645
Number of discrete wavelet transform levels.
646646
calc_peak_gpu_mem: str:
647-
Supporting parameter for memory calculation in HTTomo.
647+
Parameter to support memory estimation in HTTomo. Irrelevant to the method itself and can be ignored by user.
648648
649649
Returns
650650
-------

tests/test_prep/test_stripe.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,9 @@ def test_remove_stripe_fw_on_data(data, flats, darks):
5757
data_norm = dark_flat_field_correction(data, flats, darks, cutoff=10)
5858
data_norm = minus_log(data_norm)
5959

60-
data_after_stripe_removal = remove_stripe_fw(cp.copy(data_norm), wname='sym16', sigma=1, level=7).get()
60+
data_after_stripe_removal = remove_stripe_fw(
61+
cp.copy(data_norm), wname="sym16", sigma=1, level=7
62+
).get()
6163

6264
assert_allclose(np.mean(data_after_stripe_removal), 0.279236, rtol=1e-05)
6365
assert_allclose(

0 commit comments

Comments
 (0)