Fix import error (#3554)

Fixes an import error in PR #3453.
NVIDIA · Dec 10, 2024 · 608868f · 608868f
1 parent 875d765
commit 608868f
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/benchmarks/python/test_transpose.py b/benchmarks/python/test_transpose.py
@@ -4,14 +4,9 @@
 import pytest
 from nvfuser import FusionDefinition, DataType
 from nvfuser.pytorch_utils import torch_dtype_to_nvfuser_dtype
-from .core import run_benchmark, clear_dynamo_cache
+from .core import run_benchmark, clear_dynamo_cache, with_executor
 import torch
-from .global_params import (
-    generate_input_sizes,
-    FLOAT_DTYPES,
-    PROMOTE_DTYPES,
-    with_executor,
-)
+from .global_params import generate_input_sizes, FLOAT_DTYPES, PROMOTE_DTYPES
 
 
 def transpose_fusion(