chore(sdk): fix error 'ApiResourceReference' object has no attribute …

…'id' (#6997) * chore(sdk): fix error 'ApiResourceReference' object has no attribute 'id' * fix more
kubeflow · Dec 3, 2021 · bc7aad8 · bc7aad8
1 parent 636503c
commit bc7aad8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sdk/python/kfp/cli/pipeline.py b/sdk/python/kfp/cli/pipeline.py
@@ -214,7 +214,7 @@ def _print_pipeline_versions(versions: List[kfp_server_api.ApiPipelineVersion],
         version.created_at.isoformat(),
         next(rr
              for rr in version.resource_references
-             if rr.key.type == kfp_server_api.ApiResourceType.PIPELINE).id
+             if rr.key.type == kfp_server_api.ApiResourceType.PIPELINE).key.id
     ]
             for version in versions]
     print_output(data, headers, output_format, table_format="grid")
@@ -252,7 +252,7 @@ def _display_pipeline_version(version: kfp_server_api.ApiPipelineVersion,
                               output_format: OutputFormat):
     pipeline_id = next(
         rr for rr in version.resource_references
-        if rr.key.type == kfp_server_api.ApiResourceType.PIPELINE).id
+        if rr.key.type == kfp_server_api.ApiResourceType.PIPELINE).key.id
     table = [["Pipeline ID", pipeline_id], ["Version name", version.name],
              ["Uploaded at", version.created_at.isoformat()],
              ["Version ID", version.id]]

diff --git a/sdk/python/kfp/cli/run.py b/sdk/python/kfp/cli/run.py
@@ -222,7 +222,7 @@ def _print_runs(runs: List[kfp_server_api.ApiRun], output_format: OutputFormat):
         run.created_at.isoformat(),
         next(rr
              for rr in run.resource_references
-             if rr.key.type == kfp_server_api.ApiResourceType.EXPERIMENT).id
+             if rr.key.type == kfp_server_api.ApiResourceType.EXPERIMENT).key.id
     ]
             for run in runs]
     print_output(data, headers, output_format, table_format='grid')