如何修复TypeError: load()缺少一个必需的位置参数: Colab中的“加载程序”错误？

Question

我想在科拉布经营这个笔记本。

from openai.embeddings_utils import get_embedding

# This will take just under 10 minutes
df['babbage_similarity'] = df.combined.apply(lambda x: get_embedding(x, engine='text-similarity-babbage-001'))
df['babbage_search'] = df.combined.apply(lambda x: get_embedding(x, engine='text-search-babbage-doc-001'))
df.to_csv('/content/documents.csv')

当我试图运行它时，这个代码块会失败--我得到这个错误：

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-31-b04fdb9f95d8> in <module>()
      5
      6 
----> 7 from openai.embeddings_utils import get_embedding
      8 
      9 # This will take just under 10 minutes

10 frames
/usr/local/lib/python3.7/dist-packages/distributed/config.py in <module>()
     18 
     19 with open(fn) as f:
---> 20     defaults = yaml.load(f)
     21 
     22 dask.config.update_defaults(defaults)

TypeError: load() missing 1 required positional argument: 'Loader'

在谷歌搜索的基础上，这似乎是针对Colab的版本问题。但我没能修好它。有什么想法吗？

Answer 1

在config.py中

代之以：

with open(fn) as f:
    defaults = yaml.load(f)

在这方面：

with open(fn) as f:
    defaults = yaml.safe_load(f)