Skip to content

Commit 12924ce

Browse files
committed
fix lint.
1 parent a4760ec commit 12924ce

File tree

3 files changed

+7
-3
lines changed

3 files changed

+7
-3
lines changed

src/ptwt/continuous_transform.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ def _integrate(
196196
if type(arr) is np.ndarray:
197197
integral = np.cumsum(arr)
198198
elif type(arr) is torch.Tensor:
199-
integral = torch.cumsum(arr, -1) # type: ignore
199+
integral = torch.cumsum(arr, -1) # type: ignore
200200
else:
201201
raise TypeError("Only ndarrays or tensors are integratable.")
202202
integral *= step

src/ptwt/packets.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,11 @@ def _get_wavedec(
177177
return self._matrix_wavedec_dict[length]
178178
else:
179179
return partial(
180-
wavedec, wavelet=self.wavelet, level=1, mode=self.mode, axis=self.axis # type: ignore
180+
wavedec,
181+
wavelet=self.wavelet,
182+
level=1,
183+
mode=self.mode,
184+
axis=self.axis, # type: ignore
181185
)
182186

183187
def _get_waverec(

src/ptwt/separable_conv_transform.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ def _separable_conv_waverecn(
165165

166166
approx: torch.Tensor = coeffs[0]
167167
for level_dict in coeffs[1:]:
168-
keys = list(level_dict.keys()) # type: ignore
168+
keys = list(level_dict.keys()) # type: ignore
169169
level_dict["a" * max(map(len, keys))] = approx # type: ignore
170170
approx = _separable_conv_idwtn(level_dict, wavelet) # type: ignore
171171
return approx

0 commit comments

Comments
 (0)