Skip to content

Commit cf5bb20

Browse files
peri044peri044
committed
chore: Linter fixes
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent d49b46c commit cf5bb20

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

70 files changed

+279
-70
lines changed

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_adaptive_avgpool.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestAdaptiveAvgPoolConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_avgpool.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import param, parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestAvgPoolConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_batchnorm.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,10 @@
11
import torch
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from torch.testing._internal.common_utils import run_tests
4-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
4+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
5+
AccTestCase,
6+
InputTensorSpec,
7+
)
58

69

710
class TestBatchNormConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_binary_ops.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,10 @@
66
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
77
from parameterized import parameterized
88
from torch.testing._internal.common_utils import run_tests
9-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
9+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
10+
AccTestCase,
11+
InputTensorSpec,
12+
)
1013

1114
NEED_TEST_BOTH_CONSTANTS_CASE = True
1215

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_cat.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,10 @@
33
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
44
from parameterized import param, parameterized
55
from torch.testing._internal.common_utils import run_tests
6-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
6+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
7+
AccTestCase,
8+
InputTensorSpec,
9+
)
710

811

912
class TestCatConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_chunk.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,10 @@
33
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
44
from parameterized import parameterized
55
from torch.testing._internal.common_utils import run_tests
6-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
6+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
7+
AccTestCase,
8+
InputTensorSpec,
9+
)
710

811

912
class TestChunkConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_clamp.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import param, parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestClampConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_convolution.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import param, parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestConvolutionConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_dequantize.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,10 @@
66

77
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
88
from torch.testing._internal.common_utils import run_tests
9-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
9+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
10+
AccTestCase,
11+
InputTensorSpec,
12+
)
1013

1114

1215
@unittest.skip(

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_einsum.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,10 @@
33
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
44
from parameterized import parameterized
55
from torch.testing._internal.common_utils import run_tests
6-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
6+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
7+
AccTestCase,
8+
InputTensorSpec,
9+
)
710

811

912
class TestConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_elu.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch.nn as nn
33
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestELUConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_embedding.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,10 @@
55
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
66
from parameterized import param, parameterized
77
from torch.testing._internal.common_utils import run_tests
8-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
8+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
9+
AccTestCase,
10+
InputTensorSpec,
11+
)
912

1013

1114
@unittest.skip(

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_eq.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestEqConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_gelu.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,10 @@
44
import torch.nn as nn
55
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
66
from torch.testing._internal.common_utils import run_tests
7-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
7+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
8+
AccTestCase,
9+
InputTensorSpec,
10+
)
811

912

1013
@unittest.skip(

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_getitem.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,10 @@
33
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
44
from parameterized import parameterized
55
from torch.testing._internal.common_utils import run_tests
6-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
6+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
7+
AccTestCase,
8+
InputTensorSpec,
9+
)
710

811

912
class TestGetitemConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_gt.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestGtConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_hard_sigmoid.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
from parameterized import parameterized
33
from torch import nn
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69
from torch_tensorrt.fx.tracer.acc_tracer import acc_ops
710

811

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_hardtanh.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,10 @@
33
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
44
from parameterized import parameterized
55
from torch.testing._internal.common_utils import run_tests
6-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
6+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
7+
AccTestCase,
8+
InputTensorSpec,
9+
)
710

811

912
class TestHardtanhConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_interpolate.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,10 @@
33
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
44
from parameterized import parameterized
55
from torch.testing._internal.common_utils import run_tests
6-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
6+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
7+
AccTestCase,
8+
InputTensorSpec,
9+
)
710

811

912
class TestInterpolateConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_isinf.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,10 @@
44

55
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
66
from torch.testing._internal.common_utils import run_tests
7-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
7+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
8+
AccTestCase,
9+
InputTensorSpec,
10+
)
811

912

1013
@unittest.skip("Implementation is commented out due to accuracy issue T113156424")

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_leaky_relu.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch.nn as nn
33
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestLeakyReLUConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_linear.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestLinearConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_logical_and.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestAndMethodSimpleConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_logical_or.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestLogicalOrMethodSimpleConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_logical_xor.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestLogicalXorMethodSimpleConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_lt.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestLtConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_matmul.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,10 @@
33
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
44
from parameterized import parameterized
55
from torch.testing._internal.common_utils import run_tests
6-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
6+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
7+
AccTestCase,
8+
InputTensorSpec,
9+
)
710

811

912
class TestMatMulConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_max.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestMaxConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_maximum.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,10 @@
11
import torch
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from torch.testing._internal.common_utils import run_tests
4-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
4+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
5+
AccTestCase,
6+
InputTensorSpec,
7+
)
58

69

710
class TestMaximumConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_maxpool.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import param, parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestMaxPoolConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_min.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
5+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
6+
AccTestCase,
7+
InputTensorSpec,
8+
)
69

710

811
class TestMinConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_minimum.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,10 @@
11
import torch
22
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
33
from torch.testing._internal.common_utils import run_tests
4-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
4+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
5+
AccTestCase,
6+
InputTensorSpec,
7+
)
58

69

710
class TestMinimumConverter(AccTestCase):

‎py/torch_tensorrt/dynamo/fx_ts_compat/test/converters/acc_op/test_narrow.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,10 @@
33
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
44
from parameterized import parameterized
55
from torch.testing._internal.common_utils import run_tests
6-
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import AccTestCase, InputTensorSpec
6+
from torch_tensorrt.dynamo.fx_ts_compat.tools.common_fx2trt import (
7+
AccTestCase,
8+
InputTensorSpec,
9+
)
710

811

912
class TestNarrowConverterWithDynamicShape(AccTestCase):

0 commit comments

Comments
 (0)