Skip to content

Commit b9b7f63

Browse files
committed
chore: Fix linting
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 867e956 commit b9b7f63

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/py/test_qat_trt_accuracy.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,9 @@ def test_compile_script(self):
5454
log(Level.Info, "[Pyt FP32] Test Acc: {:.2f}%".format(100 * fp32_test_acc))
5555

5656
compile_spec = {
57-
"inputs": [trtorch.Input([16, 3, 32, 32])],
58-
"op_precision": torch.int8,
59-
# "enabled_precision": {torch.float32, torch.int8},
57+
"inputs": [trtorch.Input([16, 3, 32, 32])],
58+
"op_precision": torch.int8,
59+
# "enabled_precision": {torch.float32, torch.int8},
6060
}
6161

6262
trt_mod = trtorch.compile(self.model, compile_spec)

0 commit comments

Comments
 (0)