diff options
author | 2018-05-10 11:57:20 -0700 | |
---|---|---|
committer | 2018-05-10 12:00:17 -0700 | |
commit | 3ffa132c03ff02decc86a31d8bf888e9381278a7 (patch) | |
tree | 136099708056150061c0fce5bdaae676177b1daa /tensorflow/python/kernel_tests/distributions/BUILD | |
parent | 71b88284d9834f83a5d73feda3cf67944b878362 (diff) |
Use distribution_util.arguments instead of locals. This fixes a bug in newer python version
where locals is a dynamic list.
PiperOrigin-RevId: 196150149
Diffstat (limited to 'tensorflow/python/kernel_tests/distributions/BUILD')
-rw-r--r-- | tensorflow/python/kernel_tests/distributions/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/kernel_tests/distributions/BUILD b/tensorflow/python/kernel_tests/distributions/BUILD index f3cc9636f9..cf2e8832fd 100644 --- a/tensorflow/python/kernel_tests/distributions/BUILD +++ b/tensorflow/python/kernel_tests/distributions/BUILD @@ -41,6 +41,7 @@ cuda_py_test( "//tensorflow/python:math_ops", "//tensorflow/python:platform_test", ], + shard_count = 3, ) cuda_py_test( |