TFX企鹅教程penguin_tfdv.ipynb出现TypeCheckError问题求助

运行未修改的TFX官方企鹅教程penguin_tfdv.ipynb时出现以下错误：

TypeCheckError: Output type hint violation at WriteStatsOutput[train]: expected <class 'apache_beam.pvalue.PDone'>, got <class 'str'>
Full type hint:
IOTypeHints[inputs=((<class 'tensorflow_metadata.proto.v0.statistics_pb2.DatasetFeatureStatisticsList'>,), {}), outputs=((<class 'apache_beam.pvalue.PDone'>,), {})]

解决方案 #

• 锁定Apache Beam兼容版本
  该错误源于TFX与Apache Beam的版本不兼容，执行以下命令卸载现有版本并安装指定兼容版本：

  !pip uninstall -y apache-beam
  !pip install apache-beam==2.46.0
  

• 重启运行时
  安装完成后，重启Colab运行时（菜单栏「运行时」→「重启运行时」），再重新执行所有代码单元格。

原因说明 #

官方教程的代码基于特定版本的TFX和Apache Beam编写，高版本Beam会改变WriteStatsOutput组件的返回值类型，导致类型检查失败，锁定兼容版本即可解决该问题。

内容的提问来源于stack exchange，提问作者Ovtchinnikov