# make sure that memory profiler does not hang on exception
from memory_profiler import memory_usage

def foo():
    raise NotImplementedError('Error')

try:
    out = memory_usage((foo, tuple(), {}), timeout=1)
except NotImplementedError:
    pass
print('Success')