From 0e77f3603953546e351f7b337307d73182c8342c Mon Sep 17 00:00:00 2001
From: Doojin Park <doojin@google.com>
Date: Mon, 2 Dec 2024 14:02:23 +0000
Subject: [PATCH] Add xfail to the flaky tests

---
 tfx/components/statistics_gen/executor_test.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tfx/components/statistics_gen/executor_test.py b/tfx/components/statistics_gen/executor_test.py
index 2acf4ac474..272489147e 100644
--- a/tfx/components/statistics_gen/executor_test.py
+++ b/tfx/components/statistics_gen/executor_test.py
@@ -14,6 +14,7 @@
 """Tests for tfx.components.statistics_gen.executor."""
 
 import os
+import pytest
 import tempfile
 
 from absl.testing import parameterized
@@ -288,6 +289,7 @@ def testDoWithSchemaAndStatsOptions(self):
             "expected_sample_rate_by_split_property": {"train": 0.2, "eval": 0.2},
         },
     )
+    @pytest.mark.xfail(run=False, reason="Flaky test")
     def testDoWithSamplingProperty(
         self, sample_rate, sample_rate_by_split, expected_sample_rate_by_split_property
     ):