From b9b7f632570080439f7d0b82499a47db730abf94 Mon Sep 17 00:00:00 2001 From: Dheeraj Peri Date: Tue, 10 Aug 2021 23:34:26 -0700 Subject: [PATCH] chore: Fix linting Signed-off-by: Dheeraj Peri --- tests/py/test_qat_trt_accuracy.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/py/test_qat_trt_accuracy.py b/tests/py/test_qat_trt_accuracy.py index 06e18fdf2a..3620bcca1e 100644 --- a/tests/py/test_qat_trt_accuracy.py +++ b/tests/py/test_qat_trt_accuracy.py @@ -54,9 +54,9 @@ def test_compile_script(self): log(Level.Info, "[Pyt FP32] Test Acc: {:.2f}%".format(100 * fp32_test_acc)) compile_spec = { - "inputs": [trtorch.Input([16, 3, 32, 32])], - "op_precision": torch.int8, - # "enabled_precision": {torch.float32, torch.int8}, + "inputs": [trtorch.Input([16, 3, 32, 32])], + "op_precision": torch.int8, + # "enabled_precision": {torch.float32, torch.int8}, } trt_mod = trtorch.compile(self.model, compile_spec)