Skip to content

Commit

Permalink
[Fix] Support YAML files in kubeflow v2 dag runner (#6808)
Browse files Browse the repository at this point in the history
* 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>
  • Loading branch information
vkarampudi and tfx-copybara authored May 13, 2024
1 parent 98c58ee commit 3ab5da5
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
3 changes: 2 additions & 1 deletion RELEASE.md
Original file line number Diff line number Diff line change
Expand Up @@ -19,4 +19,5 @@
| -- | -- | -- | -- |
| `kfp-pipeline-spec` | `kfp-pipeline-spec>=0.1.10,<0.2` | `>0.1.13,<0.2` | |

## Documentation Updates
## Documentation Updates

4 changes: 2 additions & 2 deletions tfx/orchestration/kubeflow/v2/kubeflow_v2_dag_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)


Expand Down

0 comments on commit 3ab5da5

Please sign in to comment.