Skip to content

Commit

Permalink
Shard distribution_layer_test so it doesn't time out as often.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 263012414
axch authored and tensorflower-gardener committed Aug 12, 2019
1 parent 0666f73 commit 8d1400f
Showing 2 changed files with 2 additions and 0 deletions.
1 change: 1 addition & 0 deletions tensorflow_probability/python/layers/BUILD
Original file line number Diff line number Diff line change
@@ -138,6 +138,7 @@ py_test(
name = "distribution_layer_test",
size = "large",
srcs = ["distribution_layer_test.py"],
shard_count = 5,
deps = [
# numpy dep,
# tensorflow dep,
Original file line number Diff line number Diff line change
@@ -80,6 +80,7 @@ def setUp(self):
self.train_size, *self.input_shape) > 0.75).astype(np.float32)
self.x_test = (np.random.rand(
self.test_size, *self.input_shape) > 0.75).astype(np.float32)
super(EndToEndTest, self).setUp()

def test_keras_sequential_api(self):
"""Test `DistributionLambda`s are composable via Keras `Sequential` API."""

0 comments on commit 8d1400f

Please sign in to comment.