Skip to content

pickle

util.pickle

handle_unpickling_error 

handle_unpickling_error(function)
Source code in SaigeToolkit/util/pickle.py 
 7
 8
 9
10
11
12
13
14
15
def handle_unpickling_error(function):
    @wraps(function)
    def wrapper(*args, **kwargs):
        try:
            return function(*args, **kwargs)
        except pickle.UnpicklingError:
            raise InvalidModelFileError

    return wrapper