Skip to content

Commit

Permalink
fix lint.
Browse files Browse the repository at this point in the history
  • Loading branch information
v0lta committed Jul 28, 2023
1 parent dbed76d commit f44800e
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
2 changes: 1 addition & 1 deletion src/ptwt/matmul_transform_2.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,9 @@
from .conv_transform import _get_filter_tensors
from .conv_transform_2 import (
_check_if_tensor,
_construct_2d_filt,
_wavedec2d_unfold_channels_2d_list,
_waverec2d_fold_channels_2d_list,
_construct_2d_filt,
)
from .matmul_transform import construct_boundary_a, construct_boundary_s, orthogonalize
from .sparse_math import (
Expand Down
4 changes: 3 additions & 1 deletion tests/test_convolution_fwt_3.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,9 @@ def _cat_batch_list(batch_lists: List) -> List:
)
@pytest.mark.parametrize("wavelet", ["haar", "db2", "db4"])
@pytest.mark.parametrize("level", [1, 2, None])
@pytest.mark.parametrize("mode", ["reflect", "zero", "constant", "periodic", "symmetric"])
@pytest.mark.parametrize(
"mode", ["reflect", "zero", "constant", "periodic", "symmetric"]
)
def test_waverec3(shape: list, wavelet: str, level: int, mode: str) -> None:
"""Ensure the 3d analysis transform is invertible."""
data = np.random.randn(*shape)
Expand Down

0 comments on commit f44800e

Please sign in to comment.