fix(sdk): fix yaml support create_run_from_pipeline_func (#7500)

kubeflow · Apr 1, 2022 · 9e708b9 · 9e708b9
1 parent 48644ed
commit 9e708b9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sdk/python/kfp/client/client.py b/sdk/python/kfp/client/client.py
@@ -993,8 +993,7 @@ def create_run_from_pipeline_func(
         ).strftime('%Y-%m-%d %H-%M-%S')
 
         with tempfile.TemporaryDirectory() as tmpdir:
-            # TODO: support IR yaml
-            pipeline_package_path = os.path.join(tmpdir, 'pipeline.json')
+            pipeline_package_path = os.path.join(tmpdir, 'pipeline.yaml')
             compiler.Compiler().compile(
                 pipeline_func=pipeline_func,
                 package_path=pipeline_package_path,