Fix PEP8.

thu-ml · Jul 21, 2020 · 3f89363 · 3f89363
1 parent f793320
commit 3f89363
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/base/test_batch.py b/test/base/test_batch.py
@@ -36,9 +36,9 @@ def test_batch():
     with pytest.raises(TypeError):
         Batch(a=[torch.zeros((2, 3)), torch.zeros((3, 3))])
     with pytest.raises(TypeError):
-        Batch(a=[torch.zeros(3,3), np.zeros([3,3])])
+        Batch(a=[torch.zeros((3, 3)), np.zeros((3, 3))])
     with pytest.raises(TypeError):
-        Batch(a=[1, np.zeros([3,3]), torch.zeros(3,3)])
+        Batch(a=[1, np.zeros((3, 3)), torch.zeros((3, 3))])
     batch = Batch(a=[torch.ones(3), torch.ones(3)])
     assert torch.allclose(batch.a, torch.ones(2, 3))
     batch = Batch(obs=[0], np=np.zeros([3, 4]))