Re-enable distribution tests since we had a workaround.

PiperOrigin-RevId: 338137775 Change-Id: If9af5ca659b402591f20f35e5e9ed9cf9e630ff2
2020-10-20 14:25:06 -07:00 · 2020-10-20 14:25:06 -07:00 · a7aa84c9c6
commit a7aa84c9c6
parent 5832fa274f
1 changed files with 0 additions and 4 deletions
--- a/tensorflow/python/kernel_tests/distributions/BUILD
+++ b/tensorflow/python/kernel_tests/distributions/BUILD
@ -114,10 +114,6 @@ cuda_py_test(
    name = "dirichlet_test",
    size = "small",
    srcs = ["dirichlet_test.py"],
-    tags = [
-        # b/170982175
-        "no_oss",
-    ],
    deps = [
        "//tensorflow/python:client_testlib",
        "//tensorflow/python:framework_for_generated_wrappers",