chore: Fix linting

Signed-off-by: Dheeraj Peri <[email protected]>
pytorch · Aug 11, 2021 · b9b7f63 · b9b7f63
1 parent 867e956
commit b9b7f63
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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)