diff --git a/test/registered/distributed/test_disaggregation_hybrid_attention.py b/test/registered/distributed/test_disaggregation_hybrid_attention.py index a926894ef..c004a31f0 100644 --- a/test/registered/distributed/test_disaggregation_hybrid_attention.py +++ b/test/registered/distributed/test_disaggregation_hybrid_attention.py @@ -12,9 +12,7 @@ from sglang.test.test_utils import ( popen_launch_pd_server, ) -register_cuda_ci( - est_time=400, suite="stage-c-test-8-gpu-h200", disabled="TCP fallback flaky" -) +register_cuda_ci(est_time=500, suite="stage-c-test-8-gpu-h200") @unittest.skipIf(is_in_ci(), "Temporarily disable the flaky test.") @@ -158,10 +156,6 @@ 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."""