message: Increase the size of the flaky sample_stats_test.py
diff: @@ -341,7 +341,7 @@ cuda_py_test(
  cuda_py_test(
  name = "sample_stats_test",
  - size = "small",
  + size = "medium",
  srcs = ["python/kernel_tests/sample_stats_test.py"],
  additional_deps = [
  ":distributions_py",
  