Fix a TPU test failure.

PiperOrigin-RevId: 317935580 Change-Id: Id0d06446f4d7159b375879ea97f251d10ee80195
2020-06-23 14:03:51 -07:00 · 2020-06-23 14:03:51 -07:00 · cc9d951afa
commit cc9d951afa
parent 68fe003b85
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/python/keras/layers/preprocessing/BUILD
+++ b/tensorflow/python/keras/layers/preprocessing/BUILD
@ -369,6 +369,7 @@ distribute_py_test(
    tags = [
        "multi_and_single_gpu",
    ],
+    tpu_tags = ["no_oss"],
    deps = [
        ":category_crossing",
        "//tensorflow/python/distribute:combinations",