hashiruAI / CEO /utils /suppress_outputs.py
Harshil Patel
add budget manager and bug fixes
8cf77a3
raw
history blame
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