Fix Mamba2Metadata dropping has_mamba_track_mask (#26877)

2026-06-01 22:03:59 +08:00
parent fd16e05252
commit 693adabff7
2 changed files with 2 additions and 1 deletions
@@ -316,7 +316,6 @@ class TestDisaggregationHybridAttentionMamba(PDDisaggregationServerBase):
        self.assertGreater(metrics["score"], 0.87)


-@unittest.skipIf(is_in_ci(), "Temporarily disable the flaky test.")
 class TestDisaggregationHybridAttentionMambaExtraBuffer(PDDisaggregationServerBase):
    @classmethod
    def setUpClass(cls):