模块导入泡菜行为

Question

我要保存一个依赖于名为foo的模块的sklearn估计器。

脚本是这样的：

from foo import *
import pickle
import sklearn

estimator = ...

pickle.dump(estimator, open('file', 'wb'))

但是，当尝试在另一个脚本中重新加载该估算器时：

import pickle
estimator = pickle.load(open('file', 'rb'))

我收到以下错误： ModuleNotFoundError: No module named `foo`

虽然实际上不需要foo，但我只是用它来构建估算器，但是要用它来进行预测，例如，不需要foo。

关于如何解决该问题的任何想法？

我试图使用莳萝库并将函数保存在foo中的相同的pickle文件中，但仍然出现错误。