_pickle.PicklingError：无法腌制<class'__main __。x'=“”>：__main__上的属性查找X失败

Question

腌汁在Jupiter笔记本中完全成功，但是当我在.py文件中腌制失败。

我不明白为什么？

这只是一个简单的课程

class X:
def __init__(self, ref):
    import torch
    self.ref = torch.tensor(ref)
    self.result = torch.full(ref.shape, True, dtype=torch.uint8)
    self.index = None

def partial_fit(self, X):
    assert len(X) > 0
    import numpy
    blablabla

def transform(self, X):
    return X[:, self.index]

Python版本和软件包看起来是同一版本