ComfyUI/comfy/component_model/suppress_stdout.py
2025-05-07 14:53:39 -07:00

15 lines
356 B
Python

import contextlib
import io
import sys
@contextlib.contextmanager
def suppress_stdout_stderr():
new_stdout, new_stderr = io.StringIO(), io.StringIO()
old_stdout, old_stderr = sys.stdout, sys.stderr
try:
sys.stdout, sys.stderr = new_stdout, new_stderr
yield
finally:
sys.stdout, sys.stderr = old_stdout, old_stderr