Skip to content

Commit 7449b2b

Browse files
committed
format fix
1 parent 3eed4e3 commit 7449b2b

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

tests/python/relax/test_frontend_from_fx.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2725,7 +2725,9 @@ def forward(self, input):
27252725
@tvm.script.ir_module
27262726
class expected1:
27272727
@R.function
2728-
def main(input_1: R.Tensor((1, 3, 10, 10), dtype="float32")) -> R.Tuple(
2728+
def main(
2729+
input_1: R.Tensor((1, 3, 10, 10), dtype="float32")
2730+
) -> R.Tuple(
27292731
R.Tensor((1, 1, 10, 10), dtype="float32"),
27302732
R.Tensor((1, 1, 10, 10), dtype="float32"),
27312733
R.Tensor((1, 1, 10, 10), dtype="float32"),
@@ -2784,7 +2786,9 @@ def forward(self, data):
27842786
@tvm.script.ir_module
27852787
class expected1:
27862788
@R.function
2787-
def main(input_1: R.Tensor((3, 3, 10, 10), dtype="float32")) -> R.Tuple(
2789+
def main(
2790+
input_1: R.Tensor((3, 3, 10, 10), dtype="float32")
2791+
) -> R.Tuple(
27882792
R.Tensor((3, 10, 10), dtype="float32"),
27892793
R.Tensor((3, 10, 10), dtype="float32"),
27902794
R.Tensor((3, 10, 10), dtype="float32"),
@@ -2819,7 +2823,9 @@ def main(input_1: R.Tensor((3, 3, 10, 10), dtype="float32")) -> R.Tuple(
28192823
@tvm.script.ir_module
28202824
class expected2:
28212825
@R.function
2822-
def main(input_1: R.Tensor((3, 3, 10, 10), dtype="float32")) -> R.Tuple(
2826+
def main(
2827+
input_1: R.Tensor((3, 3, 10, 10), dtype="float32")
2828+
) -> R.Tuple(
28232829
R.Tensor((3, 10, 10), dtype="float32"),
28242830
R.Tensor((3, 10, 10), dtype="float32"),
28252831
R.Tensor((3, 10, 10), dtype="float32"),
@@ -2888,7 +2894,9 @@ def forward(self, input):
28882894
@tvm.script.ir_module
28892895
class Expected:
28902896
@R.function
2891-
def main(input_1: R.Tensor((1, 3, 10, 10), dtype="float32")) -> R.Tuple(
2897+
def main(
2898+
input_1: R.Tensor((1, 3, 10, 10), dtype="float32")
2899+
) -> R.Tuple(
28922900
R.Tensor((1, 1, 10, 10), dtype="float32"),
28932901
R.Tensor((1, 1, 10, 10), dtype="float32"),
28942902
R.Tensor((1, 1, 10, 10), dtype="float32"),

0 commit comments

Comments
 (0)