Skip to content

Commit 0235dbe

Browse files
committed
refactor: apply linting
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent d38eee7 commit 0235dbe

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/py/test_api.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,7 @@ def test_compile_script(self):
7373
same = (trt_mod(self.input) - self.scripted_model(self.input)).abs().max()
7474
self.assertTrue(same < 2e-2)
7575

76+
7677
class TestCompileHalf(ModelTestCase):
7778

7879
def setUp(self):
@@ -94,6 +95,7 @@ def test_compile_script_half(self):
9495
same = (trt_mod(self.input.half()) - self.scripted_model(self.input.half())).abs().max()
9596
self.assertTrue(same < 2e-2)
9697

98+
9799
class TestCompileHalfDefault(ModelTestCase):
98100

99101
def setUp(self):
@@ -115,6 +117,7 @@ def test_compile_script_half_by_default(self):
115117
same = (trt_mod(self.input.half()) - self.scripted_model(self.input.half())).abs().max()
116118
self.assertTrue(same < 2e-2)
117119

120+
118121
class TestFallbackToTorch(ModelTestCase):
119122

120123
def setUp(self):

0 commit comments

Comments
 (0)