frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix usages of contextmanager without finally (#96170)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/96170
Approved by: https://github.com/ngimel, https://github.com/malfet
This commit is contained in:
Horace He
2023-03-07 23:27:01 +00:00
committed by PyTorch MergeBot
parent 34d18c8bee
commit 5bbec680d7
15 changed files with 91 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
torch/serialization.py
View File

@ -56,8 +56,10 @@ class SourceChangeWarning(Warning):
@contextmanager
def mkdtemp():
path = tempfile.mkdtemp()
yield path
shutil.rmtree(path)
try:
yield path
finally:
shutil.rmtree(path)
_package_registry = []