Enforce black formatting.

Author: Talmaj

onnx2pytorch/operations/expand.py

@@ -4,6 +4,5 @@
 
 class Expand(nn.Module):
     def forward(self, input: torch.Tensor, shape: torch.Tensor):
-        #return input.expand(torch.Size(shape))
+        # return input.expand(torch.Size(shape))
         return input * torch.ones(torch.Size(shape), dtype=input.dtype)
-

onnx2pytorch/operations/squeeze.py

@@ -10,7 +10,7 @@ def __init__(self, opset_version, dim=None):
         self.dim = dim
         super().__init__()
 
-    def forward(self, input: torch.Tensor, axes: torch.Tensor=None):
+    def forward(self, input: torch.Tensor, axes: torch.Tensor = None):
         if self.opset_version < 13:
             dims = self.dim
         else:

onnx2pytorch/operations/unsqueeze.py

@@ -3,13 +3,14 @@
 
 from onnx2pytorch.operations.base import Operator
 
+
 class Unsqueeze(Operator):
     def __init__(self, opset_version, dim=None):
         self.opset_version = opset_version
         self.dim = dim
         super().__init__()
 
-    def forward(self, data: torch.Tensor, axes: torch.Tensor=None):
+    def forward(self, data: torch.Tensor, axes: torch.Tensor = None):
         if self.opset_version < 13:
             dims = self.dim
         else:
@@ -22,4 +23,3 @@ def forward(self, data: torch.Tensor, axes: torch.Tensor=None):
             for dim in sorted(dims, reverse=True):
                 data = torch.unsqueeze(data, dim=dim)
             return data
-

tests/onnx2pytorch/operations/test_expand.py

@@ -3,6 +3,7 @@
 
 from onnx2pytorch.operations.expand import Expand
 
+
 def test_expand_dim_changed():
     op = Expand()
     inp = torch.reshape(torch.arange(0, 3, dtype=torch.float32), [3, 1])
@@ -12,6 +13,7 @@ def test_expand_dim_changed():
     assert tuple(op(inp, new_shape).shape) == exp_shape
     assert torch.equal(op(inp, new_shape), exp)
 
+
 def test_expand_dim_unchanged():
     op = Expand()
     inp = torch.reshape(torch.arange(0, 3, dtype=torch.int32), [3, 1])

tests/onnx2pytorch/operations/test_range.py

@@ -3,6 +3,7 @@
 
 from onnx2pytorch.operations.range import Range
 
+
 @pytest.mark.parametrize(
     "start, limit, delta, expected",
     [

tests/onnx2pytorch/operations/test_squeeze.py

@@ -23,6 +23,7 @@ def test_squeeze_v11(inp, dim, exp_shape):
     op = Squeeze(opset_version=11, dim=dim)
     assert tuple(op(inp).shape) == exp_shape
 
+
 @pytest.mark.parametrize(
     "dim, exp_shape",
     [