From 33443d083e129d61f857bc0563ed1b7f8c8fab0b Mon Sep 17 00:00:00 2001 From: Junjie Bai Date: Tue, 16 Apr 2019 21:44:22 -0700 Subject: [PATCH] Fix python lint (#19331) Summary: VitalyFedyunin jerryzh168 Pull Request resolved: https://github.com/pytorch/pytorch/pull/19331 Differential Revision: D14969435 Pulled By: bddppq fbshipit-source-id: c1555c52064758ecbe668f92b837f2d7524f6118 --- test/test_torch.py | 2 +- torch/onnx/symbolic.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_torch.py b/test/test_torch.py index e7d6193..bbad85a 100644 --- a/test/test_torch.py +++ b/test/test_torch.py @@ -2693,7 +2693,7 @@ class _TestTorchMixin(object): rqr = qr.dequantize() print(r.numpy()) print(rqr.numpy()) - self.assertTrue(np.allclose(r.numpy(), rqr.numpy(), atol=2/scale)) + self.assertTrue(np.allclose(r.numpy(), rqr.numpy(), atol=2 / scale)) @unittest.skipIf(torch.cuda.device_count() < 2, 'fewer than 2 GPUs detected') def test_device_guard(self): diff --git a/torch/onnx/symbolic.py b/torch/onnx/symbolic.py index 75f3062..567af64 100644 --- a/torch/onnx/symbolic.py +++ b/torch/onnx/symbolic.py @@ -1348,7 +1348,7 @@ def ones_like(g, input, dtype, layout, device, pin_memory=False): def full(g, sizes, value, dtype, layout, device, pin_memory=False): - if pin_memory and _parse_arg(pin_memory,'b'): + if pin_memory and _parse_arg(pin_memory, 'b'): raise RuntimeError("onnx pin_memory support is not implemented") const_value = _maybe_get_const(value, 't') if _is_value(const_value): -- 2.7.4