Add custom_path property & update path property in kfp sdk Artifact class.

Signed-off-by: agoins <[email protected]>

Author: alyssacgoins

sdk/python/kfp/dsl/types/artifact_types.py

@@ -82,6 +82,7 @@ def __init__(self,
         self.uri = uri or ''
         self.name = name or ''
         self.metadata = metadata or {}
+        self.custom_path: str = ''
 
     @property
     def path(self) -> str:
@@ -92,6 +93,8 @@ def path(self, path: str) -> None:
         self._set_path(path)
 
     def _get_path(self) -> Optional[str]:
+        if self.custom_path != "":
+            return self.custom_path
         if self.uri.startswith(RemotePrefix.GCS.value):
             return _GCS_LOCAL_MOUNT_PREFIX + self.uri[len(RemotePrefix.GCS.value
                                                          ):]
@@ -111,6 +114,16 @@ def _get_path(self) -> Optional[str]:
     def _set_path(self, path: str) -> None:
         self.uri = convert_local_path_to_remote_path(path)
 
+    @property
+    def custom_path(self) -> str:
+        return self.custom_path
+
+    @custom_path.setter
+    def custom_path(self, value):
+        self._set_custom_path = value
+
+    def _set_custom_path(self, path: str) -> None:
+        self.custom_path = path
 
 def convert_local_path_to_remote_path(path: str) -> str:
     if path.startswith(_GCS_LOCAL_MOUNT_PREFIX):