Добавить декоратор в декоратор компонентов в KFP v2 в Vertex AI

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

Проблема в том, что когда я компилирую конвейер, я вижу, что в консоли печатается «запуск добавленной функциональности», но «запуск оболочки» не отображается в журнале в Vertex AI. Я делаю что-то неправильно?

Answer 1

Это не так. В настоящее время это разочаровывающее ограничение Kubeflow.