anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

[Enhancement] Patch OmniStage.try_collect() with _proc alive checks (#1560)

Browse Source 
Signed-off-by: Daniel Huang <daniel1.huang@intel.com>
This commit is contained in:
Daniel Huang authored and GitHub committed 2026-03-12 10:40:42 +08:00
1 parent e1efcb7560
commit ac17431745
2 files changed
+9 -3
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/entrypoints/test_omni_diffusion.py
+1 -1
View File
@@ -345,7 +345,7 @@ def _setup_multiprocessing_mocks(monkeypatch: pytest.MonkeyPatch, mocker: Mocker
fake_process_instance = mocker.MagicMock()
fake_process_instance.start = mocker.MagicMock()
fake_process_instance.join = mocker.MagicMock()
fake_process_instance.is_alive = mocker.MagicMock(return_value=False)
fake_process_instance.is_alive = mocker.MagicMock(return_value=True)
fake_process_instance.terminate = mocker.MagicMock()
fake_process_class.return_value = fake_process_instance