diff options
author | 2018-07-26 11:55:25 -0700 | |
---|---|---|
committer | 2018-07-26 11:59:14 -0700 | |
commit | 7e4d0d13e51faee4469808d577591f6cb73bbcc7 (patch) | |
tree | 85431150cd1074769e48f0847fc956afc0d4d457 /tensorflow | |
parent | 91ff408ecd52dd167d966c9df222e840f1d43f8f (diff) |
shard some estimator tests
PiperOrigin-RevId: 206196742
Diffstat (limited to 'tensorflow')
-rw-r--r-- | tensorflow/python/estimator/BUILD | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/python/estimator/BUILD b/tensorflow/python/estimator/BUILD index fd46163050..817c8e6848 100644 --- a/tensorflow/python/estimator/BUILD +++ b/tensorflow/python/estimator/BUILD @@ -171,6 +171,7 @@ py_test( name = "baseline_test", size = "medium", srcs = ["canned/baseline_test.py"], + shard_count = 4, srcs_version = "PY2AND3", tags = [ "no_pip", @@ -207,6 +208,7 @@ py_test( name = "boosted_trees_test", size = "medium", srcs = ["canned/boosted_trees_test.py"], + shard_count = 2, srcs_version = "PY2AND3", tags = [ "optonly", @@ -676,6 +678,7 @@ py_test( name = "keras_test", size = "large", srcs = ["keras_test.py"], + shard_count = 4, srcs_version = "PY2AND3", tags = [ "no_windows", |