Add xfail to the flaky tests (#7724)

tensorflow · Dec 3, 2024 · e3fe5b1 · e3fe5b1
1 parent f42957d
commit e3fe5b1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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
     ):