feat(sdk): add workaround for google artifact (support custom artifact types pt. 4) (#8279)

sdk/python/kfp/components/executor.py

@@ -331,7 +331,11 @@ def create_artifact_instance(
 
     artifact_cls = artifact_types._SCHEMA_TITLE_TO_TYPE.get(
         schema_title, artifact_cls)
-    return artifact_cls(
+    return artifact_cls._from_executor_fields(
+        uri=runtime_artifact.get('uri', ''),
+        name=runtime_artifact.get('name', ''),
+        metadata=runtime_artifact.get('metadata', {}),
+    ) if hasattr(artifact_cls, '_from_executor_fields') else artifact_cls(
         uri=runtime_artifact.get('uri', ''),
         name=runtime_artifact.get('name', ''),
         metadata=runtime_artifact.get('metadata', {}),

sdk/python/kfp/components/executor_test.py

@@ -1007,10 +1007,19 @@ class VertexDataset:
     schema_title = 'google.VertexDataset'
     schema_version = '0.0.0'
 
-    def __init__(self, name: str, uri: str, metadata: dict) -> None:
-        self.name = name
-        self.uri = uri
-        self.metadata = metadata
+    @classmethod
+    def _from_executor_fields(
+        cls,
+        name: str,
+        uri: str,
+        metadata: dict,
+    ) -> 'VertexDataset':
+
+        instance = VertexDataset()
+        instance.name = name
+        instance.uri = uri
+        instance.metadata = metadata
+        return instance
 
     @property
     def path(self) -> str:
@@ -1132,7 +1141,7 @@ class TestDictToArtifact(parameterized.TestCase):
         self.assertIsInstance(
             executor.create_artifact_instance(runtime_artifact), expected_type)
 
-    def test_dict_to_artifact_nonkfp_artifact(self):
+    def test_dict_to_artifact_google_artifact(self):
         runtime_artifact = {
             'metadata': {},
             'name': 'input_artifact_one',