Skip to content

Commit 6504bc5

Browse files
SiegeLordExjburnim
authored andcommitted
Shard student_t_process_test to make it less flaky.
Looks like lots of small test cases adding up to a lot, so sharding seems appropriate. PiperOrigin-RevId: 546381129
1 parent fc5dd37 commit 6504bc5

File tree

1 file changed

+1
-0
lines changed
  • tensorflow_probability/python/distributions

1 file changed

+1
-0
lines changed

tensorflow_probability/python/distributions/BUILD

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4485,6 +4485,7 @@ multi_substrate_py_test(
44854485
name = "student_t_process_test",
44864486
srcs = ["student_t_process_test.py"],
44874487
jax_size = "medium",
4488+
shard_count = 2,
44884489
deps = [
44894490
":multivariate_student_t",
44904491
":student_t",

0 commit comments

Comments
 (0)