diff --git a/sdk/python/kfp/compiler_cli_tests/test_data/pipeline_with_task_final_status_yaml.json b/sdk/python/kfp/compiler_cli_tests/test_data/pipeline_with_task_final_status_yaml.json index 2b720865ee2e..dc205a79aec8 100644 --- a/sdk/python/kfp/compiler_cli_tests/test_data/pipeline_with_task_final_status_yaml.json +++ b/sdk/python/kfp/compiler_cli_tests/test_data/pipeline_with_task_final_status_yaml.json @@ -66,7 +66,7 @@ "pipeline status:", "{{$.inputs.parameters['status']}}" ], - "image": "python:3.9" + "image": "python:3.7" } }, "exec-print-op": { @@ -75,7 +75,7 @@ "echo", "{{$.inputs.parameters['message']}}" ], - "image": "python:3.9" + "image": "python:3.7" } } } @@ -143,4 +143,4 @@ }, "schemaVersion": "2.1.0", "sdkVersion": "kfp-1.8.11" -} \ No newline at end of file +} diff --git a/sdk/python/kfp/compiler_cli_tests/test_data/pipeline_with_task_final_status_yaml.py b/sdk/python/kfp/compiler_cli_tests/test_data/pipeline_with_task_final_status_yaml.py index 0ad48c874787..5734394dce67 100644 --- a/sdk/python/kfp/compiler_cli_tests/test_data/pipeline_with_task_final_status_yaml.py +++ b/sdk/python/kfp/compiler_cli_tests/test_data/pipeline_with_task_final_status_yaml.py @@ -13,9 +13,9 @@ # limitations under the License. """Pipeline using ExitHandler with PipelineTaskFinalStatus (YAML).""" +from kfp import compiler from kfp import components from kfp import dsl -from kfp import compiler from kfp.dsl import component exit_op = components.load_component_from_text(""" @@ -25,7 +25,7 @@ - {name: status, type: PipelineTaskFinalStatus} implementation: container: - image: python:3.9 + image: python:3.7 command: - echo - "user input:" @@ -40,7 +40,7 @@ - {name: message, type: String} implementation: container: - image: python:3.9 + image: python:3.7 command: - echo - {inputValue: message}