[CI] Fix test_communication.py and add port cleanup (#7021)

This commit is contained in:
YuBaoku
2026-03-26 10:56:40 +08:00
committed by GitHub
parent e6804ba97d
commit 61ebac49ef
5 changed files with 25 additions and 2 deletions
+1 -1
View File
@@ -52,7 +52,7 @@ class TestCommunicationBasic(unittest.TestCase):
communication.use_custom_allreduce()
self.assertIsNotNone(communication._TP_AR)
mock_custom_ar.assert_called_once_with(fake_group, 64 * 1024 * 1024)
mock_custom_ar.assert_called_once_with(fake_group, 8 * 1024 * 1024)
def test_custom_ar_clear_ipc_handles(self):
mock_tp_ar = MagicMock()