GPy/GPy/util/decorators.py
2013-03-12 16:46:35 +00:00

16 lines
453 B
Python

import numpy as np
from functools import wraps
def silence_errors(f):
"""
This wraps a function and it silences numpy errors that
happen during the execution. After the function has exited, it restores
the previous state of the warnings.
"""
@wraps(f)
def wrapper(*args, **kwds):
status = np.seterr(all='ignore')
result = f(*args, **kwds)
np.seterr(**status)
return result
return wrapper