fix: fix Calibration.from_pretrained support for pathlib.Path

pyannote · Feb 4, 2025 · 27bf53c · 27bf53c
1 parent bb6bc1e
commit 27bf53c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyannote/audio/core/calibration.py b/pyannote/audio/core/calibration.py
@@ -113,7 +113,7 @@ def from_file(cls, path: str) -> "Calibration":
     @classmethod
     def from_pretrained(
         cls,
-        checkpoint: str,
+        checkpoint: Union[Path, str],
         subfolder: Optional[str] = None,
         token: Optional[Text] = None,
         cache_dir: Optional[Union[str, Path]] = None,
@@ -138,7 +138,7 @@ def from_pretrained(
             return cls.from_file(checkpoint)
 
         _, _, path = download_from_hf_hub(
-            checkpoint,
+            str(checkpoint),
             AssetFileName.Calibration,
             subfolder=subfolder,
             cache_dir=cache_dir,