missed a spot w/ earlier lintrunner.

NVIDIA · Feb 5, 2025 · 0ac84f6 · 0ac84f6
1 parent dfe3392
commit 0ac84f6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/benchmarks/python/test_cat.py b/benchmarks/python/test_cat.py
@@ -402,7 +402,9 @@ def benchmark_fn(inputs):
 
 
 @pytest.mark.parametrize("executor", ["thunder", "torchcompile"])
-def test_cat_phi3_1_baseline_benchmark(benchmark, executor: str, disable_validation: bool) -> None:
+def test_cat_phi3_1_baseline_benchmark(
+    benchmark, executor: str, disable_validation: bool
+) -> None:
     def to_be_compiled(t14):
         # t14: "cuda:0 f32[1, 48, 2048]"
         t0 = torch.permute(t14, (0, 2, 1))  # t0: "cuda:0 f32[1, 2048, 48]"