From dac37215840e55f5024294c379dc6e01d5ea4257 Mon Sep 17 00:00:00 2001 From: Ken Franko <kfranko@google.com> Date: Thu, 27 Feb 2020 00:54:08 -0800 Subject: [PATCH] Remove TODO since workaround makes sense. PiperOrigin-RevId: 297535246 Change-Id: I4bb3006bd42b0d637d6517049b7e6f16c4f4e63b --- .../integration_tests/distribution_strategy_utils.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow/examples/saved_model/integration_tests/distribution_strategy_utils.py b/tensorflow/examples/saved_model/integration_tests/distribution_strategy_utils.py index 1501e61d475..eabe4cf3802 100644 --- a/tensorflow/examples/saved_model/integration_tests/distribution_strategy_utils.py +++ b/tensorflow/examples/saved_model/integration_tests/distribution_strategy_utils.py @@ -32,8 +32,9 @@ _strategies = [ strategy_combinations.tpu_strategy, ] -# TODO(b/145386854): The presence of GPU strategies upsets TPU initialization, -# despite their test instances being skipped early on. +# The presence of GPU strategies upsets TPU initialization, +# despite their test instances being skipped early. This is a workaround +# for b/145386854. if "test_tpu" in sys.argv[0]: _strategies = [s for s in _strategies if "GPU" not in str(s)]