precommit v2

Author: JanFSchulte

test/pytest/test_pytorch_api.py

@@ -75,7 +75,6 @@ def test_linear(backend, io_type):
 )
 @pytest.mark.parametrize('backend', ['Vivado', 'Vitis', 'Quartus'])
 @pytest.mark.parametrize('io_type', ['io_parallel', 'io_stream'])
-
 def test_activations(activation_function, backend, io_type):
     model = torch.nn.Sequential(nn.Linear(1, 1), activation_function).to()
     model.eval()
@@ -120,12 +119,14 @@ def __init__(self):
     def forward(self, x):
         return nn.functional.relu(x)
 
+
 class SoftmaxModel(nn.Module):
     def __init__(self):
         super().__init__()
 
     def forward(self, x):
-        return nn.functional.softmax(x,dim=-1)
+        return nn.functional.softmax(x, dim=-1)
+
 
 class TanHModel(nn.Module):
     def __init__(self):