kubeflow
/
pipelines
mirror of https://github.com/kubeflow/pipelines.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test: fix sdk execution test (#9988) 

* fix sdk execution test

* fix missing key

* fix state

* fix state

* fix capitalization
This commit is contained in:
Chen Sun 2023-09-14 13:02:30 -07:00 committed by GitHub
parent fc1f12b7bd
commit a13e143038
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
sdk/python/test_data/test_data_config.yaml
View File

@ -239,6 +239,7 @@ components:
execute: false
- module: component_with_metadata_fields
name: dataset_joiner
execute: false
- module: component_with_task_final_status
name: exit_comp
execute: false

4
test/sdk-execution-tests/sdk_execution_tests.py
View File

@ -65,7 +65,7 @@ def create_test_case_parameters() -> List[TestCase]:
return parameters
def wait(run_result: client.client.RunPipelineResult) -> kfp_server_api.ApiRun:
def wait(run_result: client.client.RunPipelineResult) -> kfp_server_api.V2beta1Run:
return kfp_client.wait_for_run_completion(
run_id=run_result.run_id, timeout=int(TIMEOUT_SECONDS))
@ -122,4 +122,4 @@ async def test(test_case: TestCase, mocker) -> None:
f'Error triggering pipeline {test_case.name}.') from e
api_run = await event_loop.run_in_executor(None, wait, run_result)
assert api_run.run.status == 'Succeeded', f'Pipeline {test_case.name} ended with incorrect status: {api_run.run.status}. More info: {run_url}'
assert api_run.state == 'SUCCEEDED', f'Pipeline {test_case.name} ended with incorrect status: {api_run.state}. More info: {run_url}'