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)]