append_to_stem: Use to_arr

Setsugennoao · Jun 3, 2024 · 6624b4f · 6624b4f
1 parent 21c5309
commit 6624b4f
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/stgpytools/types/file.py b/stgpytools/types/file.py
@@ -4,6 +4,8 @@
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Callable, Iterable, Literal, TypeAlias, Union
 
+from ..functions import to_arr
+
 __all__ = [
     'FilePathType', 'FileDescriptor',
     'FileOpener',
@@ -102,10 +104,7 @@ def write_lines(
     def append_to_stem(self, suffixes: str | Iterable[str], sep: str = "_") -> SPath:
         """Append a suffix to the stem of the path"""
 
-        if isinstance(suffixes, str):
-            suffixes = [suffixes]
-
-        return self.with_stem(sep.join([self.stem, *suffixes]))
+        return self.with_stem(sep.join([self.stem, *to_arr(suffixes)]))  # type:ignore[list-item]
 
 
 SPathLike = Union[str, Path, SPath]