diff options
| author | orbit-stabilizer <33772483+orbit-stabilizer@users.noreply.github.com> | 2017-11-19 17:02:05 -0800 |
|---|---|---|
| committer | Eric Wieser <wieser.eric@gmail.com> | 2017-11-19 17:02:05 -0800 |
| commit | eac60563f4a6fda56403870854e786ef718caf8f (patch) | |
| tree | 18bc8fba27fb9e04aa9cdc2fcad7e0b9a34acf5e /numpy/ma | |
| parent | 37cf9141b1e1647128b584137743821283548548 (diff) | |
| download | numpy-eac60563f4a6fda56403870854e786ef718caf8f.tar.gz | |
BUG: Have np.ma.dump and np.ma.load close their files (#10055)
Fixes #10045
Diffstat (limited to 'numpy/ma')
| -rw-r--r-- | numpy/ma/core.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/numpy/ma/core.py b/numpy/ma/core.py index 0d6702790..a10d354ee 100644 --- a/numpy/ma/core.py +++ b/numpy/ma/core.py @@ -7818,8 +7818,10 @@ def dump(a, F): """ if not hasattr(F, 'readline'): - F = open(F, 'w') - return pickle.dump(a, F) + with open(F, 'w') as F: + pickle.dump(a, F) + else: + pickle.dump(a, F) def dumps(a): @@ -7859,8 +7861,10 @@ def load(F): """ if not hasattr(F, 'readline'): - F = open(F, 'r') - return pickle.load(F) + with open(F, 'r') as F: + pickle.load(F) + else: + pickle.load(F) def loads(strg): |