[F] Fixed testing files error

Author: zephyr-sh

tests/heatmap_reg/test_hmap_infer.py

@@ -1,5 +1,6 @@
 import numpy as np
 import pytest
+
 from docaligner.heatmap_reg.infer import Inference
 
 
@@ -14,7 +15,7 @@ def mock_onnx_engine(mocker):
     return mock_engine
 
 
-@pytest.mark.parametrize("model_cfg", ['lcnet050', 'lcnet050_fpn', 'lcnet100', 'lcnet100_fpn', 'mobilenetv2_140', 'fastvit_t8', 'fastvit_sa24'])
+@pytest.mark.parametrize("model_cfg", ['lcnet100', 'fastvit_t8', 'fastvit_sa24'])
 def test_inference_initialization(model_cfg):
     inference = Inference(model_cfg=model_cfg)
     assert inference.cfg == Inference.configs[model_cfg]

tests/point_reg/test_point_infer.py

@@ -1,5 +1,6 @@
 import numpy as np
 import pytest
+
 from docaligner.point_reg.infer import Inference