Reformat using Black.

Author: fkodom

benchmark.py

@@ -41,16 +41,32 @@ def benchmark_conv(
 
     torch_conv = {1: f.conv1d, 2: f.conv2d, 3: f.conv3d}[signal.ndim - 2]
     direct_time = benchmark(
-        torch_conv, signal, kernel, bias=bias, padding=padding, stride=stride, groups=groups
+        torch_conv,
+        signal,
+        kernel,
+        bias=bias,
+        padding=padding,
+        stride=stride,
+        groups=groups,
     )
     fourier_time = benchmark(
-        fft_conv, signal, kernel, bias=bias, padding=padding, stride=stride, groups=groups
+        fft_conv,
+        signal,
+        kernel,
+        bias=bias,
+        padding=padding,
+        stride=stride,
+        groups=groups,
     )
     print(f"Direct time: {direct_time}")
     print(f"Fourier time: {fourier_time}")
 
-    y0 = torch_conv(signal, kernel, bias=bias, padding=padding, stride=stride, groups=groups)
-    y1 = fft_conv(signal, kernel, bias=bias, padding=padding, stride=stride, groups=groups)
+    y0 = torch_conv(
+        signal, kernel, bias=bias, padding=padding, stride=stride, groups=groups
+    )
+    y1 = fft_conv(
+        signal, kernel, bias=bias, padding=padding, stride=stride, groups=groups
+    )
     abs_error = torch.abs(y0 - y1)
     print(f"Output size: {y0.size()}")
     print(f"Abs Error Mean: {abs_error.mean():.3E}")