[请教]关于pickle中的使用

pickle的文档里好像提到说,在load的时候,pickle只能解析模块根目录下的类,而类里面的子类则无法读取。
我一个程序中dump了一个包含子类的类,然后我发现无法用pickle.load来读取,抛出的异常显示getattr(mod, name)函数调用无法找到某个子类

我查了文档,好像有个关于Subclassing Unpickler的内容,不过看不明白,不知所云。

后来我通过子类继承Unpickler,并重载find_class函数,代码如下:
class KCUnpickler(Unpickler):
    def find_class(self, mod, name):
        try:
            return Unpickler.find_class(self, mod, name)
        except:
            return getattr(KernelComponent, name)


这样我就可以从文件中读取我的类了。。不知道正确的做法是什么?