hashiruAI / src /manager /utils /suppress_outputs.py
helloparthshah's picture
refactor
2f85c93
raw
history blame contribute delete
452 Bytes
def suppress_output(func):
"""
Decorator to suppress output of a function.
"""
def wrapper(*args, **kwargs):
import sys
import os
sys.stdout = open(os.devnull, 'w')
sys.stderr = open(os.devnull, 'w')
try:
result = func(*args, **kwargs)
finally:
sys.stdout = sys.__stdout__
sys.stderr = sys.__stderr__
return result
return wrapper