fix import error

NVIDIA · Dec 10, 2024 · fdf49b5 · fdf49b5
1 parent 0a23556
commit fdf49b5
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(