google / sedpack / 11778864763

89%

push

github

Implement Rust generator for from_generator (#44)

Implement Rust geneartor for tf.data.Dataset.from_generator. The
function tf.data.Dataset.from_generator is taking a callable returning
an iterator. This can get called multiple times even when the iterator
does not end (raise an StopIteration). This supports being reentrant
(see https://github.com/google/sedpack/pull/33).

---------

Co-authored-by: kralka <karelkral@google.com>

41 of 46 new or added lines in 1 file covered. (89.13%)

2313 of 2676 relevant lines covered (86.43%)

0.86 hits per line