06_visualization_pipeline fails execution with v2

[strangiato]

06_visualization_pipeline fails when executing with v2.

Here are the full execution logs:

I0510 23:54:27.117913      51 launcher_v2.go:90] input ComponentSpec:{
  "outputDefinitions": {
    "artifacts": {
      "confusion_matrix_path": {
        "artifactType": {
          "schemaTitle": "system.Artifact",
          "schemaVersion": "0.0.1"
        }
      },
      "mlpipeline_ui_metadata_path": {
        "artifactType": {
          "schemaTitle": "system.Artifact",
          "schemaVersion": "0.0.1"
        }
      }
    }
  },
  "executorLabel": "exec-confusion-matrix-viz"
}
I0510 23:54:27.119007      51 cache.go:116] Connecting to cache endpoint 172.31.222.7:8887
[KFP Executor 2024-05-10 23:54:28,909 INFO]: Looking for component `confusion_matrix_viz` in --component_module_path `/tmp/tmp.HWfTI7ELBR/ephemeral_component.py`
[KFP Executor 2024-05-10 23:54:28,909 INFO]: Loading KFP component "confusion_matrix_viz" from /tmp/tmp.HWfTI7ELBR/ephemeral_component.py (directory "/tmp/tmp.HWfTI7ELBR" and module name "ephemeral_component")
[KFP Executor 2024-05-10 23:54:28,910 INFO]: Got executor_input:
{
    "inputs": {},
    "outputs": {
        "artifacts": {
            "confusion_matrix_path": {
                "artifacts": [
                    {
                        "type": {
                            "schemaTitle": "system.Artifact",
                            "schemaVersion": "0.0.1"
                        },
                        "uri": "s3://pipelines/metadata-example-pipeline/99d1a8f9-48a2-418d-96c6-65fba1a7e8b2/confusion-matrix-viz/confusion_matrix_path"
                    }
                ]
            },
            "mlpipeline_ui_metadata_path": {
                "artifacts": [
                    {
                        "type": {
                            "schemaTitle": "system.Artifact",
                            "schemaVersion": "0.0.1"
                        },
                        "uri": "s3://pipelines/metadata-example-pipeline/99d1a8f9-48a2-418d-96c6-65fba1a7e8b2/confusion-matrix-viz/mlpipeline_ui_metadata_path"
                    }
                ]
            }
        },
        "outputFile": "/tmp/kfp_outputs/output_metadata.json"
    }
}
Traceback (most recent call last):
  File "<frozen runpy>", line 198, in _run_module_as_main
  File "<frozen runpy>", line 88, in _run_code
  File "/opt/app-root/lib64/python3.11/site-packages/kfp/dsl/executor_main.py", line 109, in <module>
    executor_main()
  File "/opt/app-root/lib64/python3.11/site-packages/kfp/dsl/executor_main.py", line 101, in executor_main
    output_file = executor.execute()
                  ^^^^^^^^^^^^^^^^^^
  File "/opt/app-root/lib64/python3.11/site-packages/kfp/dsl/executor.py", line 362, in execute
    return self.write_executor_output(result)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/app-root/lib64/python3.11/site-packages/kfp/dsl/executor.py", line 268, in write_executor_output
    raise RuntimeError(
RuntimeError: Unknown return type: <class 'inspect._empty'>. Must be one of `str`, `int`, `float`, a subclass of `Artifact`, or a NamedTuple collection of these types.
I0510 23:54:28.933298      51 launcher_v2.go:151] publish success.
F0510 23:54:28.933348      51 main.go:49] failed to execute component: exit status 1
time="2024-05-10T23:54:29.096Z" level=info msg="sub-process exited" argo=true error="<nil>"
Error: exit status 1
time="2024-05-10T23:54:30.062Z" level=info msg="sub-process exited" argo=true error="<nil>"
Error: exit status 1