Error de archivo de importación de pickle AttributeError: No se puede obtener el atributo'XXX 'en el módulo ＜' __ main__ 'del procesamiento

class CustomUnpickler(pickle.Unpickler):

    def find_class(self, module, name):
        try:
            return super().find_class(__name__