[Fix] Support YAML files in kubeflow v2 dag runner (#6808)

* Support pipeline spec 2.1 with backward compatibility PiperOrigin-RevId: 631251911 * Support YAML files in kubeflow v2 dag runner PiperOrigin-RevId: 631970915 * Update version.py for 1.15.1 Release * Automated rollback of commit 9d86d1a PiperOrigin-RevId: 632488503 * Support YAML files in kubeflow v2 dag runner PiperOrigin-RevId: 633095531 --------- Co-authored-by: tfx-team <tensorflow-extended-nonhuman@googlegroups.com>
tensorflow · May 13, 2024 · 3ab5da5 · 3ab5da5
1 parent 98c58ee
commit 3ab5da5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/RELEASE.md b/RELEASE.md
@@ -19,4 +19,5 @@
 | -- | -- | -- | -- |
 | `kfp-pipeline-spec` | `kfp-pipeline-spec>=0.1.10,<0.2` | `>0.1.13,<0.2` | |
 
-## Documentation Updates
+## Documentation Updates
+
diff --git a/tfx/orchestration/kubeflow/v2/kubeflow_v2_dag_runner.py b/tfx/orchestration/kubeflow/v2/kubeflow_v2_dag_runner.py
@@ -89,12 +89,12 @@ def _write_pipeline_spec_to_file(
     yaml_comments = _extract_comments_from_pipeline_spec(
         pipeline_spec_dict, pipeline_description
     )
-    with open(package_path, 'w') as yaml_file:
+    with fileio.open(package_path, 'w') as yaml_file:
       yaml_file.write(yaml_comments)
       documents = [pipeline_spec_dict]
       yaml.dump_all(documents, yaml_file, sort_keys=True)
   else:
-    with open(package_path, 'w') as json_file:
+    with fileio.open(package_path, 'w') as json_file
       json.dump(pipeline_job_dict, json_file, sort_keys=True)