From a9e706fb10cf050a60883880fd40b03e0d048c6b Mon Sep 17 00:00:00 2001 From: Vicent Caselles Date: Tue, 29 Jul 2025 14:54:35 +0200 Subject: [PATCH] convert crop_slices to tuple --- fft_conv_pytorch/fft_conv.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/fft_conv_pytorch/fft_conv.py b/fft_conv_pytorch/fft_conv.py index 81a181e..3f59e6f 100644 --- a/fft_conv_pytorch/fft_conv.py +++ b/fft_conv_pytorch/fft_conv.py @@ -132,10 +132,12 @@ def fft_conv( output = irfftn(output_fr, dim=tuple(range(2, signal.ndim))) # Remove extra padded values - crop_slices = [slice(None), slice(None)] + [ - slice(0, (signal_size[i] - kernel.size(i) + 1), stride_[i - 2]) - for i in range(2, signal.ndim) - ] + crop_slices = tuple( + [slice(None), slice(None)] + [ + slice(0, (signal_size[i] - kernel.size(i) + 1), stride_[i - 2]) + for i in range(2, signal.ndim) + ] + ) output = output[crop_slices].contiguous() # Optionally, add a bias term before returning.