mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
[CI] Fix test_communication.py and add port cleanup (#7021)
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user