提交 efad433e authored 作者: Zach Ploskey's avatar Zach Ploskey

ensure opened files are closed

上级 4d4fd6d8
...@@ -152,7 +152,10 @@ def run(stdout, stderr, argv, theano_nose, batch_size, time_profile, ...@@ -152,7 +152,10 @@ def run(stdout, stderr, argv, theano_nose, batch_size, time_profile,
stderr.flush() stderr.flush()
assert rval == 0 assert rval == 0
noseids_file = '.noseids' noseids_file = '.noseids'
data = cPickle.load(open(noseids_file, 'rb'))
with open(noseids_file, 'rb') as f:
data = cPickle.load(f)
ids = data['ids'] ids = data['ids']
n_tests = len(ids) n_tests = len(ids)
if n_tests == 0: if n_tests == 0:
...@@ -193,8 +196,9 @@ def run(stdout, stderr, argv, theano_nose, batch_size, time_profile, ...@@ -193,8 +196,9 @@ def run(stdout, stderr, argv, theano_nose, batch_size, time_profile,
# otherwise this field may get erased. We use a set because it # otherwise this field may get erased. We use a set because it
# seems like it is not systematically erased though, and we want # seems like it is not systematically erased though, and we want
# to avoid duplicates. # to avoid duplicates.
failed = failed.union(cPickle.load(open(noseids_file, 'rb')) with open(noseids_file, 'rb') as f:
['failed']) failed = failed.union(cPickle.load(f)['failed'])
print '%s%% done in %.3fs (failed: %s)' % ( print '%s%% done in %.3fs (failed: %s)' % (
(test_range[-1] * 100) // n_tests, t1 - t0, len(failed)) (test_range[-1] * 100) // n_tests, t1 - t0, len(failed))
# Sort for cosmetic purpose only. # Sort for cosmetic purpose only.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论