python - cPickle 的导入错误

Question

我浏览了相关 cPickle 问题的各种回复，但似乎没有一个有帮助。使用 cloudpickle 代替 cPickle 也没有效果。

我有一个名为 MyModule 的模块，它定义了一个 MyClass 类

首先，我在与项目文件夹不同的文件夹中运行脚本。我从项目中导入我自己的类 MyClass，创建该类的实例并在目标路径中对其进行 pickle。

sys.path.append(os.path.join(os.environ.get("PROJECT_ROOT"), 'src/'))
from MyModule import MyClass
class_instance = MyClass()
with gzip.open(os.path.join(target_path, 'net.p.gz'), "wb") as f:
    cPickle.dump(class_instance, f)

然后，在我的主项目中，我想加载 pickled 文件

from MyModule import MyClass
with gzip.open(os.path.join(os.environ['PROJECT_ROOT'], 'resources/net.p.gz'), 'rb') as f:
        class_instance = cPickle.load(f)

这会导致

ImportError: No module named MyModule

但是

test_instance = MyClass()
print (test_instance)

在同一文件中打印

<MyModule.MyClass object at 0x7fad03e3ead0>

所以该模块显然已经存在并且正在被改进。

两次我都指的是同一个模块，它没有改变位置。我错过了什么？

Answer 1

假设您在这里有 MyModule:~/work/repo/my_module.py。如果您希望 from MyModule import MyClass 正常工作，那么您必须将其放在 python 路径中:~/work/repo。尝试执行 import sys;打印 sys.path 并检查一下。