Merge pull request #1580 from leandrolcampos:betaincgrad

PiperOrigin-RevId: 464004088

Author: tensorflower-gardener

tensorflow_probability/python/math/BUILD

@@ -481,9 +481,10 @@ multi_substrate_py_test(
     name = "special_test",
     size = "medium",
     srcs = ["special_test.py"],
-    shard_count = 6,
+    shard_count = 10,
     deps = [
         # absl/testing:parameterized dep,
+        # mpmath dep,
         # numpy dep,
         # tensorflow dep,
         "//tensorflow_probability",