Skip to content

Commit 56cbaaf

Browse files
committed
linting fixes
1 parent ca00344 commit 56cbaaf

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

tests/py/dynamo/conversion/test_rand_aten.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
import torch
22
import torch.nn as nn
3+
import torch_tensorrt
34
from parameterized import parameterized
45
from torch.testing._internal.common_utils import TestCase, run_tests
5-
import torch_tensorrt
66

77
rand_ops = [
88
(
@@ -75,13 +75,14 @@ def forward(self):
7575
return self.rand_op(self.size)
7676

7777
grid_model = TestModule(op, shape_or_input)
78-
#cannot use self.run_test() since it expects input in form of tensor
79-
80-
#self.run_test(grid_model, None)
78+
# cannot use self.run_test() since it expects input in form of tensor
79+
80+
# self.run_test(grid_model, None)
8181
fx_graph = torch.fx.symbolic_trace(grid_model)
8282
torch._dynamo.reset()
8383

84-
optimized_model = torch_tensorrt.compile(fx_graph,
84+
optimized_model = torch_tensorrt.compile(
85+
fx_graph,
8586
"torch_compile",
8687
None,
8788
min_block_size=1,

0 commit comments

Comments
 (0)