From 6324e50357a8288408620a5e9cc1b3022880ef41 Mon Sep 17 00:00:00 2001 From: Wilsin Gosti Date: Mon, 11 Sep 2023 20:57:15 -0700 Subject: [PATCH] #tf-data The default value of `warm_start` should be `None` instead of `False`. Setting it to `False` as the default value may break the `tf.data.Options` propagation when users set it to `True` somewhere in the input pipeline. PiperOrigin-RevId: 564585601 --- tensorflow/python/data/ops/options.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/data/ops/options.py b/tensorflow/python/data/ops/options.py index bfc870c6af4..ef6727516f4 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( -- GitLab