Skip to content

Commit 6565e60

Browse files
authored
Merge pull request #78 from v0lta/black2024
Reformatted with Black 2024 style
2 parents a95b4b5 + 05aad35 commit 6565e60

20 files changed

+24
-8
lines changed

‎noxfile.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""This module implements our CI function calls."""
2+
23
import nox
34

45

‎src/ptwt/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Differentiable and gpu enabled fast wavelet transforms in PyTorch."""
2+
23
from ._util import Wavelet
34
from .continuous_transform import cwt
45
from .conv_transform import wavedec, waverec

‎src/ptwt/_util.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Utility methods to compute wavelet decompositions from a dataset."""
2+
23
from typing import Any, Callable, List, Optional, Protocol, Sequence, Tuple, Union
34

45
import numpy as np

‎src/ptwt/continuous_transform.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
33
This module is based on pywt's cwt implementation.
44
"""
5+
56
from typing import Any, Tuple, Union
67

78
import numpy as np

‎src/ptwt/conv_transform.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
33
This module treats boundaries with edge-padding.
44
"""
5+
56
# Created by moritz wolter, 14.04.20
67
from typing import List, Optional, Sequence, Tuple, Union
78

‎src/ptwt/conv_transform_2.py‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
torch.nn.functional.conv_transpose2d under the hood.
55
"""
66

7-
87
from functools import partial
98
from typing import List, Optional, Tuple, Union
109

‎src/ptwt/conv_transform_3.py‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,10 @@ def wavedec3(
211211

212212
def _waverec3d_fold_channels_3d_list(
213213
coeffs: List[Union[torch.Tensor, Dict[str, torch.Tensor]]],
214-
) -> Tuple[List[Union[torch.Tensor, Dict[str, torch.Tensor]]], List[int],]:
214+
) -> Tuple[
215+
List[Union[torch.Tensor, Dict[str, torch.Tensor]]],
216+
List[int],
217+
]:
215218
# fold the input coefficients for processing conv2d_transpose.
216219
fold_coeffs: List[Union[torch.Tensor, Dict[str, torch.Tensor]]] = []
217220
ds = list(_check_if_tensor(coeffs[0]).shape)

‎src/ptwt/matmul_transform.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
in Strang Nguyen (p. 32), as well as the description
77
of boundary filters in "Ripples in Mathematics" section 10.3 .
88
"""
9+
910
# Created by moritz (wolter@cs.uni-bonn.de) at 14.04.20
1011
import sys
1112
from typing import List, Optional, Union

‎src/ptwt/matmul_transform_2.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
33
This module uses boundary filters to minimize padding.
44
"""
5+
56
# Written by moritz ( @ wolter.tech ) in 2021
67
import sys
78
from functools import partial

‎src/ptwt/matmul_transform_3.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Implement 3D separable boundary transforms."""
2+
23
import sys
34
from functools import partial
45
from typing import Dict, List, NamedTuple, Optional, Tuple, Union

0 commit comments

Comments
 (0)