diff --git a/test/srt/test_disaggregation_hybrid_attention.py b/test/srt/test_disaggregation_hybrid_attention.py index c8154e94d..04b99fd9b 100644 --- a/test/srt/test_disaggregation_hybrid_attention.py +++ b/test/srt/test_disaggregation_hybrid_attention.py @@ -153,6 +153,10 @@ class TestDisaggregationHybridAttentionMambaExtraBuffer(PDDisaggregationServerBa self.assertGreater(metrics["accuracy"], 0.93) +@unittest.skipIf( + is_in_ci(), + "Temporarily disable the flaky test: tcp fallback is not stable currently.", +) class TestDisaggregationHybridAttentionMambaDPDecode(PDDisaggregationServerBase): """Test with prefill tp=2 and decode tp=2/dp=2 with dp-attention enabled."""