Make arpa_to_fst work when arguments are str (from Samsung, AI Center, Cambridge) (speechbrain#2919)

rogiervd · Rogier van Dalen · web-flow · commit 4ade5319899d · 2025-05-15T13:39:34.000+01:00
Co-authored-by: Rogier van Dalen &lt;r.vandalen@samsung.com&gt;
diff --git a/speechbrain/lm/arpa.py b/speechbrain/lm/arpa.py
@@ -325,6 +325,11 @@ def arpa_to_fst(
             "Install using `pip install kaldilm`."
         )
 
+    if isinstance(out_fst, str):
+        out_fst = Path(out_fst)
+    if isinstance(in_arpa, str):
+        in_arpa = Path(in_arpa)
+
     if cache and out_fst.exists():
         return
     if not in_arpa.exists():
