fix test cases failed in nightly pipeline (#26714)
This commit is contained in:
@@ -888,6 +888,12 @@ class Engine(EngineScoreMixin, EngineBase):
|
|||||||
and self.tokenizer_manager._subprocess_watchdog is not None
|
and self.tokenizer_manager._subprocess_watchdog is not None
|
||||||
):
|
):
|
||||||
self.tokenizer_manager._subprocess_watchdog.stop()
|
self.tokenizer_manager._subprocess_watchdog.stop()
|
||||||
|
|
||||||
|
send_to_rpc = getattr(self, "send_to_rpc", None)
|
||||||
|
if send_to_rpc is not None:
|
||||||
|
send_to_rpc.close(linger=0)
|
||||||
|
self.send_to_rpc = None
|
||||||
|
|
||||||
kill_process_tree(os.getpid(), include_parent=False, wait_timeout=60)
|
kill_process_tree(os.getpid(), include_parent=False, wait_timeout=60)
|
||||||
|
|
||||||
def __enter__(self):
|
def __enter__(self):
|
||||||
|
|||||||
@@ -37,6 +37,7 @@ class TestMiniMaxM2(GSM8KAscendMixin, CustomTestCase):
|
|||||||
"--chunked-prefill-size",
|
"--chunked-prefill-size",
|
||||||
"-1",
|
"-1",
|
||||||
]
|
]
|
||||||
|
timeout_for_server_launch = 1800
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
Reference in New Issue
Block a user