Merge branch 'saliency-debug' into 'main'

[completeness] Fix error in completeness callable

Closes #63

See merge request biosignal/software/mednet!23

src/mednet/engine/saliency/completeness.py

@@ -300,7 +300,6 @@ def run(
     else:
         raise TypeError(f"Model of type `{type(model)}` is not yet supported.")
 
-    use_cuda = device_manager.device_type == "cuda"
     if device_manager.device_type in ("cuda", "mps") and (
         parallel == 0 or parallel > 1
     ):
@@ -322,7 +321,6 @@ def run(
         saliency_map_algorithm,
         model,
         target_layers,  # type: ignore
-        use_cuda,
     )
 
     retval: dict[str, list[typing.Any]] = {}

src/mednet/engine/saliency/generator.py

@@ -66,7 +66,7 @@ def _create_saliency_map_callable(
             return pytorch_grad_cam.GradCAMElementWise(
                 model=model, target_layers=target_layers
             )
-        case "gradcam++", "gradcamplusplus":
+        case "gradcam++" | "gradcamplusplus":
             return pytorch_grad_cam.GradCAMPlusPlus(
                 model=model, target_layers=target_layers
             )