在顶点AI中的KFP v2中向组件装饰器中添加装饰器

Question

通常，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

问题是，当我编译管道时，我看到控制台上打印了‘启动添加功能’，但是在顶点AI中的日志中没有显示“启动包装器”。我做错了什么吗？

Answer 1

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