TypeCheckError:FlatMap和ParDo必须返回一个iterable

Question

我正在创建一个谷歌数据流管道使用Apache梁2. x

基本上，我有一个文本文件，每一行都包含一个英语句子。

我试图调用谷歌NLP（情感）API为每一个新的行/句子。

因此，我有一个调用NLP API的函数：

class CalculateSentiments(beam.DoFn):
  def process(self, element):
    language_client = language.Client()
    pre_text = re.sub('<[^>]*>', '', element)
    text = re.sub(r'[^\w]', ' ', pre_text)
    document = language_client.document_from_text(text)
    sentiment = document.analyze_sentiment().sentiment
    return sentiment.score

我用ParDo为每个句子调用这个函数。我假设，下面的ParDo将自动为文本文件中的每一行调用NLP情感api（基本上我不必遍历文本文件中的每一行！）

output = lines | beam.ParDo(CalculateSentiments())

output | WriteToText(known_args.output)

但在执行html" target="_blank">数据流后，我得到了这个错误：

TypeCheckError:FlatMap和ParDo必须返回一个iterable。结果被退回了。[在运行'ParDo（CalculateSentications）'时]回溯（最近一次调用）：
文件“/Users/gsattanthan/.local/lib/python2.7/site packages/apache_beam/runners/direct/executor.py”，第297行，调用计算器。进程元素（值）文件“/Users/gsattanthan/.local/lib/python2.7/site packages/apache\u beam/runners/direct/transform\u evaluator.py”，第366行，进程元素自身。跑步者进程（元素）文件“/Users/gsattanthan/.local/lib/python2.7/site packages/apache_beam/runners/common.py”，第267行，进程内self。重新发布增强（exn）文件“/Users/gsattanthan/.local/lib/python2.7/site packages/apache\u beam/runners/common.py”，第263行，进程内自我管理_dofn\u simple\u invoker（element）File“/Users/gsattanthan/.local/lib/python2.7/site packages/apache\u beam/runners/common.py”，第198行，在\u dofn\u simple\u invoker self中_进程输出（element，self.dofn\u进程（element.value））文件“/Users/gsattanthan/.local/lib/python2.7/site packages/apache\u beam/typehints/typecheck.py”，第60行，进程内返回self。包装器（self.dofn.process，args，kwargs）文件“/Users/gsattanthan/.local/lib/python2.7/site packages/apache_beam/typehints/typecheck.py”，包装器返回self的第84行_检查类型（结果）文件“/Users/gsattanthan/.local/lib/python2.7/site packages/apache\u beam/typehints/typecheck.py”，第98行，检查类型%type（输出））

我做错了什么？我使用Pardo的方式非常类似于Apache光束doco！

有什么想法吗？

Answer 1

实际上，用方括号括住返回变量可以解决这个问题。

现在它是有意义的，为什么错误说它需要是可编辑的！

返回[情绪.得分]