explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-05 21:00:19 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use tempfile.TemporaryDirectory

Browse Source
This commit is contained in:
Paul O'Leary McCann 2023-02-16 14:11:15 +09:00
parent b95123060a
commit 06ab850fcc
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/util.py
View File

@ -1028,12 +1028,11 @@ def make_tempdir() -> Generator[Path, None, None]:
YIELDS (Path): The path of the temp directory.
"""
d = Path(tempfile.mkdtemp())
yield d
try:
shutil.rmtree(str(d))
with tempfile.TemporaryDirectory() as td:
yield Path(td)
except PermissionError as e:
warnings.warn(Warnings.W091.format(dir=d, msg=e))
warnings.warn(Warnings.W091.format(dir=td, msg=e))
def is_cwd(path: Union[Path, str]) -> bool: