diff --git a/tensorflow/python/data/ops/options.py b/tensorflow/python/data/ops/options.py index bfc870c6af4f9d642112439da40abbeef3aabaeb..ef6727516f48fc1ae5e02e8a26772ba0b4b11530 100644 --- a/tensorflow/python/data/ops/options.py +++ b/tensorflow/python/data/ops/options.py @@ -600,7 +600,7 @@ class Options(options_lib.OptionsBase): "the expense of requiring more memory to hold prefetched elements " "between the time of iterator construction and usage." ), - default_factory=lambda: True if test_mode.TEST_MODE else False, + default_factory=lambda: True if test_mode.TEST_MODE else None, ) threading = options_lib.create_option(