fix: raise TypeError on wrong device type in Pipeline.to and Inferenc…

…e.to

Fixes 1397

pyannote/audio/core/inference.py

@@ -170,6 +170,11 @@ def __init__(
     def to(self, device: torch.device):
         """Send internal model to `device`"""
 
+        if not isinstance(device, torch.device):
+            raise TypeError(
+                f"`device` must be an instance of `torch.device`, got `{type(device).__name__}`"
+            )
+
         self.model.to(device)
         if self.model.specifications.powerset and not self.skip_conversion:
             self._powerset.to(device)

pyannote/audio/core/pipeline.py

@@ -324,9 +324,14 @@ def __call__(self, file: AudioFile, **kwargs):
 
         return self.apply(file, **kwargs)
 
-    def to(self, device):
+    def to(self, device: torch.device):
         """Send pipeline to `device`"""
 
+        if not isinstance(device, torch.device):
+            raise TypeError(
+                f"`device` must be an instance of `torch.device`, got `{type(device).__name__}`"
+            )
+
         for _, pipeline in self._pipelines.items():
             if hasattr(pipeline, "to"):
                 _ = pipeline.to(device)

pyannote/audio/pipelines/speaker_verification.py

@@ -80,6 +80,11 @@ def __init__(
         self.model_.to(self.device)
 
     def to(self, device: torch.device):
+        if not isinstance(device, torch.device):
+            raise TypeError(
+                f"`device` must be an instance of `torch.device`, got `{type(device).__name__}`"
+            )
+
         self.model_.to(device)
         self.device = device
         return self
@@ -255,6 +260,11 @@ def __init__(
         )
 
     def to(self, device: torch.device):
+        if not isinstance(device, torch.device):
+            raise TypeError(
+                f"`device` must be an instance of `torch.device`, got `{type(device).__name__}`"
+            )
+
         self.classifier_ = SpeechBrain_EncoderClassifier.from_hparams(
             source=self.embedding,
             savedir=f"{CACHE_DIR}/speechbrain",
@@ -415,6 +425,11 @@ def __init__(
         self.model_.to(self.device)
 
     def to(self, device: torch.device):
+        if not isinstance(device, torch.device):
+            raise TypeError(
+                f"`device` must be an instance of `torch.device`, got `{type(device).__name__}`"
+            )
+
         self.model_.to(device)
         self.device = device
         return self