Fix linting

Author: RalfG

im2deep/__main__.py

@@ -94,6 +94,7 @@ def cli(ctx, logging_level, profile, profile_name):
 
     console.print(_build_credits())
 
+
 # TODO:  Check that parameters match predict function in core
 # Implement psm_utils reading for calibration and prediction PSMLists
 @cli.command()

tests/test_model_ops.py

@@ -170,7 +170,9 @@ def test_predict_loop_single_output(self):
 
         with patch("im2deep._model_ops.track", return_value=mock_data):
             predictions = _model_ops._predict_loop(
-                model=model, data_loader=mock_data, device="cpu"  # type: ignore
+                model=model,
+                data_loader=mock_data,
+                device="cpu",  # type: ignore
             )
 
             assert isinstance(predictions, torch.Tensor)
@@ -191,7 +193,9 @@ def test_predict_loop_multi_output(self):
 
         with patch("im2deep._model_ops.track", return_value=mock_data):
             predictions = _model_ops._predict_loop(
-                model=model, data_loader=mock_data, device="cpu"  # type: ignore
+                model=model,
+                data_loader=mock_data,
+                device="cpu",  # type: ignore
             )
 
             assert isinstance(predictions, torch.Tensor)
@@ -212,7 +216,9 @@ def test_predict_loop_no_grad(self):
         # Mock track to return our mock data
         with patch("im2deep._model_ops.track", return_value=mock_data):
             predictions = _model_ops._predict_loop(
-                model=model, data_loader=mock_data, device="cpu"  # type: ignore
+                model=model,
+                data_loader=mock_data,
+                device="cpu",  # type: ignore
             )
 
             assert not predictions.requires_grad