Use TemporaryDirectory for simplicity.

jaraco · Dec 30, 2022 · ebd0910 · ebd0910
1 parent 48cc9e6
commit ebd0910
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/pip_run/ephemeral.py b/pip_run/ephemeral.py
@@ -1,13 +1,9 @@
 import contextlib
 import tempfile
-import shutil
 import pathlib
 
 
 @contextlib.contextmanager
 def context(args):
-    target = pathlib.Path(tempfile.mkdtemp(prefix='pip-run-'))
-    try:
-        yield target
-    finally:
-        shutil.rmtree(target)
+    with tempfile.TemporaryDirectory(prefix='pip-run-') as td:
+        yield pathlib.Path(td)