Fix make_initializable_iterator comment
PiperOrigin-RevId: 227598505
This commit is contained in:
parent
f0dde544f2
commit
0b6177c2fa
@ -1756,8 +1756,8 @@ def make_initializable_iterator(dataset):
|
|||||||
RuntimeError: If eager execution is enabled.
|
RuntimeError: If eager execution is enabled.
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
# Call the defined `make_one_shot_iterator()` if there is one, because some
|
# Call the defined `make_initializable_iterator()` if there is one, because
|
||||||
# datasets (e.g. for prefetching) override its behavior.
|
# some datasets (e.g. for prefetching) override its behavior.
|
||||||
return dataset.make_initializable_iterator()
|
return dataset.make_initializable_iterator()
|
||||||
except AttributeError:
|
except AttributeError:
|
||||||
return DatasetV1Adapter(dataset).make_initializable_iterator()
|
return DatasetV1Adapter(dataset).make_initializable_iterator()
|
||||||
|
Loading…
Reference in New Issue
Block a user