python - tfx.orchestration.experimental 没有属性“get_default_kubeflow_metadata_config”

Question

我正在尝试使用 kubeflow 编排 TFX 管道，但在创建管道期间出现以下错误。

AttributeError: module 'tfx.orchestration.experimental' has no attribute 'get_default_kubeflow_metadata_config'

下面是我的管道运行器文件的代码：

from tfx import v1 as tfx
import my_config
import my_pipeline1
from tfx.orchestration import experimental

def run():
  """Define a kubeflow pipeline."""

  metadata_config = experimental.get_default_kubeflow_metadata_config(
  )

  runner_config = experimental.KubeflowDagRunnerConfig(
      kubeflow_metadata_config=metadata_config,
      tfx_image=my_config.PIPELINE_IMAGE)

  experimental.KubeflowDagRunner(
      config=runner_config
  ).run(
      my_pipeline1.create_pipeline(
          pipeline_name=my_config.PIPELINE_NAME,
          pipeline_root=PIPELINE_ROOT,
          data_root=DATA_PATH,
          preprocessing_fn = my_config.PREPROCESSING_FN, 
          run_fn = my_config.RUN_FN,
      ))
if __name__ == '__main__':
  logging.set_verbosity(logging.INFO)
  run()

请让我知道是否有解决此问题的方法。我目前正在使用 TFX 1.0.0。提前致谢。

标签： pythontensorflowgoogle-cloud-platformkubeflowtfx