pre-commit fix

Author: calad0i

hls4ml/converters/pytorch/reshape.py

@@ -162,6 +162,7 @@ def handle_upsample(operation, layer_name, input_names, input_shapes, node, clas
 
     return layer, output_shape
 
+
 @pytorch_handler('ConstantPad2d')
 def parse_constantpad2d_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config):
     assert operation == 'ConstantPad2d'

test/pytest/test_pytorch_constpadmapping.py

@@ -1,9 +1,9 @@
-from pathlib import Path
-import torch
 import torch.nn as nn
+
 from hls4ml.converters import convert_from_pytorch_model
 from hls4ml.utils.config import config_from_pytorch_model
 
+
 def test_pytorch_constantpad_1d_2d():
     class Pad1DModel(nn.Module):
         def __init__(self):
@@ -22,8 +22,6 @@ def forward(self, x):
             return self.pad(x)
 
     # 1D test: batch=1, channels=2, width=4, values 1,2,3,4
-    x1d = torch.tensor([[[1., 2., 3., 4.],
-                         [4., 3., 2., 1.]]])  # shape (1, 2, 4)
     model_1d = Pad1DModel()
     model_1d.eval()
     config_1d = config_from_pytorch_model(model_1d, (2, 4))
@@ -33,8 +31,6 @@ def forward(self, x):
         print(f"{layer.name}: {layer.class_name}")
 
     # 2D test: batch=1, channels=1, height=2, width=4, values 1,2,3,4,5,6,7,8
-    x2d = torch.tensor([[[[1., 2., 3., 4.],
-                          [5., 6., 7., 8.]]]])  # shape (1, 1, 2, 4)
     model_2d = Pad2DModel()
     model_2d.eval()
     config_2d = config_from_pytorch_model(model_2d, (1, 2, 4))
@@ -45,4 +41,4 @@ def forward(self, x):
 
     # Write the HLS projects, cannot compile on Windows
     hls_model_1d.write()
-    hls_model_2d.write()
+    hls_model_2d.write()