experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make helper function global so it can be pickled.

Browse Source 
Fixes Windows tests after CL/257051579.

PiperOrigin-RevId: 257232107
This commit is contained in:
Christian Sigg 2019-07-09 11:10:53 -07:00 committed by TensorFlower Gardener
parent ca57a9d5a8
commit 3928354651
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/python/lib/io/tf_record_test.py
View File

@ -443,6 +443,10 @@ class TFRecordIteratorTest(TFCompressionTestCase):
for _ in tf_record.tf_record_iterator(fn_truncated): for _ in tf_record.tf_record_iterator(fn_truncated):
pass pass
def ChildProcess(writer, rs):
for r in rs:
writer.write(r)
writer.flush()
class TFRecordWriterCloseAndFlushTests(test.TestCase): class TFRecordWriterCloseAndFlushTests(test.TestCase):
"""TFRecordWriter close and flush tests""" """TFRecordWriter close and flush tests"""
@ -481,13 +485,8 @@ class TFRecordWriterCloseAndFlushTests(test.TestCase):
"""test Flush""" """test Flush"""
records = list(map(self._Record, range(self._num_records))) records = list(map(self._Record, range(self._num_records)))
def childProcess(writer, rs):
for r in rs:
writer.write(r)
writer.flush()
write_process = multiprocessing.Process( write_process = multiprocessing.Process(
target=childProcess, args=(self._writer, records)) target=ChildProcess, args=(self._writer, records))
write_process.start() write_process.start()
write_process.join() write_process.join()
actual = list(tf_record.tf_record_iterator(self._fn, self._options)) actual = list(tf_record.tf_record_iterator(self._fn, self._options))