Skip to content

Kubeflow V2 tests: normalize expected StatisticsGen image tag by TFX version#7818

Open
czgdp1807 wants to merge 1 commit intotensorflow:masterfrom
czgdp1807:fix_compare_test
Open

Kubeflow V2 tests: normalize expected StatisticsGen image tag by TFX version#7818
czgdp1807 wants to merge 1 commit intotensorflow:masterfrom
czgdp1807:fix_compare_test

Conversation

@czgdp1807
Copy link
Contributor

  • Updates _compare_against_testdata to normalize expected StatisticsGen_executor.container.image when golden data uses gcr.io/tfx-oss-public/tfx:latest.
  • Uses version_utils.get_image_version() so expected image tags follow the version under test.
  • Keeps expected pipeline metadata version-aware and fixes golden comparison failures in no-default-image test cases.

expected_json['pipelineSpec']['deploymentSpec'][
'executors']['StatisticsGen_executor'][
'container']['image'] = (
f'gcr.io/tfx-oss-public/tfx:{version_utils.get_image_version()}'
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Consider the following cases,

  1. If the branch being tested is master, then, version_utils.get_image_version() will give latest. Effectively no change.
  2. If the branch being tested is a release branch, r1.17.0 then, version_utils.get_image_version() will give 1.17.2. This will update latest to 1.17.2 which will pass the test. Currently without this PR this test fails on release branches.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants