在 Vertex AI 的 KFP v2 中向组件装饰器添加装饰器答案

【问题标题】：Add decorator to component decorator in KFP v2 in Vertex AI在 Vertex AI 的 KFP v2 中向组件装饰器添加装饰器
【发布时间】：2022-12-03 09:16:36
【问题描述】：

通常，KFP v2 支持添加这样的组件装饰器：

@component
def test():
  print("hello world")

我想添加一个额外的装饰器来添加这样的新功能：

@component
@added_functionality
def test():
  print("hello world")

added_functionality 被导入，看起来像这样：

from functools import wraps

def added_functionality(func):
  print("starting added functionality")

  @wraps(func)
  def wrapper(*args, **kwargs):
    print("starting wrapper")
    return func(*args, **kwargs)

  return wrapper

问题是，当我编译管道时，我看到控制台打印了“开始添加的功能”，但“开始包装器”没有显示在 Vertex AI 的日志中。难道我做错了什么？

【问题讨论】：

标签： python google-cloud-vertex-ai kfp

【解决方案1】：

你不是。这是目前 Kubeflow 的一个令人失望的限制。

【讨论】：