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

Use tempfile.TemporaryDirectory (#12285)

Browse Source
This commit is contained in:
Paul O'Leary McCann 2023-02-16 19:08:55 +09:00 committed by GitHub
parent b95123060a
commit dd3f138830
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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. YIELDS (Path): The path of the temp directory.
""" """
d = Path(tempfile.mkdtemp())
yield d
try: try:
shutil.rmtree(str(d)) with tempfile.TemporaryDirectory() as td:
yield Path(td)
except PermissionError as e: 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: def is_cwd(path: Union[Path, str]) -> bool: