diff options
author | Björn B. Brandenburg <bbb@cs.unc.edu> | 2010-03-27 11:46:13 -0400 |
---|---|---|
committer | Björn B. Brandenburg <bbb@cs.unc.edu> | 2010-03-27 11:46:13 -0400 |
commit | ebb62de88cf64ba87bcb332625cc9d75a88c7257 (patch) | |
tree | eaf7e1a1b60e1e3257ef14d3a42d00c5363b6f45 | |
parent | ede9fd4e5897cc235810facf7773260a96e18dfa (diff) |
Avoid leaking file handles.
-rw-r--r-- | util.py | 5 |
1 files changed, 4 insertions, 1 deletions
@@ -1,8 +1,11 @@ | |||
1 | import numpy as np | 1 | import numpy as np |
2 | 2 | ||
3 | def load_csv_file(fname, *args, **kargs): | 3 | def load_csv_file(fname, *args, **kargs): |
4 | return np.genfromtxt(open(fname), delimiter=",", comments="#", | 4 | f = open(fname) |
5 | data = np.genfromtxt(f, delimiter=",", comments="#", | ||
5 | *args, **kargs) | 6 | *args, **kargs) |
7 | f.close() # don't leak file handles | ||
8 | return data | ||
6 | 9 | ||
7 | def select(keep, rows): | 10 | def select(keep, rows): |
8 | ok_rows = np.zeros(len(rows), dtype=bool) | 11 | ok_rows = np.zeros(len(rows), dtype=bool) |