diff --git a/src/mednet/engine/saliency/completeness.py b/src/mednet/engine/saliency/completeness.py index a58b7e0de72fa89df06356521a5d33cceaa1c5de..197822ddf28ddc60c14aa039c457e3d4453c5652 100644 --- a/src/mednet/engine/saliency/completeness.py +++ b/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 ):