Extensions for kubeflow pipeline sdk