Fix path normalization for StreamWriter-based save operation

Follow up of pytorch#3243. Save compat module had different semantics than info and load, which requires different way of performing path normalization.
mthrok · Apr 6, 2023 · 93a68af · 93a68af
1 parent f4d94ca
commit 93a68af
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/torchaudio/_backend/utils.py b/torchaudio/_backend/utils.py
@@ -122,7 +122,7 @@ def save(
         buffer_size: int = 4096,
     ) -> None:
         save_audio(
-            os.path.normpath(uri),
+            uri,
             src,
             sample_rate,
             channels_first,

diff --git a/torchaudio/io/_stream_writer.py b/torchaudio/io/_stream_writer.py
@@ -1,3 +1,4 @@
+import os.path
 from dataclasses import dataclass
 from typing import BinaryIO, Dict, Optional, Union
 
@@ -160,12 +161,12 @@ class StreamWriter:
 
     def __init__(
         self,
-        dst: Union[str, BinaryIO],
+        dst: Union[str, BinaryIO, os.PathLike],
         format: Optional[str] = None,
         buffer_size: int = 4096,
     ):
         if isinstance(dst, str):
-            self._s = torchaudio.lib._torchaudio_ffmpeg.StreamWriter(dst, format)
+            self._s = torchaudio.lib._torchaudio_ffmpeg.StreamWriter(os.path.normpath(dst), format)
         elif hasattr(dst, "write"):
             self._s = torchaudio.lib._torchaudio_ffmpeg.StreamWriterFileObj(dst, format, buffer_size)
         else: