Skip to content

Commit 99407b9

Browse files
authored
Merge pull request #753 from NVIDIA/anuragd/py_lint
refactor: Applied py linter
2 parents 2e70c91 + b916339 commit 99407b9

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

Diff for: tests/py/test_api.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,9 @@ def test_compile_traced(self):
3232
def test_compile_script(self):
3333
with torch.no_grad():
3434
trt_mod = torchtrt.ts.compile(self.scripted_model,
35-
inputs=[self.input],
36-
device=torchtrt.Device(gpu_id=0),
37-
enabled_precisions={torch.float})
35+
inputs=[self.input],
36+
device=torchtrt.Device(gpu_id=0),
37+
enabled_precisions={torch.float})
3838
same = (trt_mod(self.input) - self.scripted_model(self.input)).abs().max()
3939
self.assertTrue(same < 2e-2)
4040

@@ -49,9 +49,9 @@ def test_compile_global(self):
4949
def test_compile_global_nn_mod(self):
5050
with torch.no_grad():
5151
trt_mod = torchtrt.compile(self.model,
52-
inputs=[self.input],
53-
device=torchtrt.Device(gpu_id=0),
54-
enabled_precisions={torch.float})
52+
inputs=[self.input],
53+
device=torchtrt.Device(gpu_id=0),
54+
enabled_precisions={torch.float})
5555
same = (trt_mod(self.input) - self.scripted_model(self.input)).abs().max()
5656
self.assertTrue(same < 2e-2)
5757

0 commit comments

Comments
 (0)