chore(sdk): Fixed the PipelineConf.data_passing_method docstring (#5217)

kubeflow · Mar 6, 2021 · 9bc63f5 · 9bc63f5
1 parent 796a5f7
commit 9bc63f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk/python/kfp/dsl/_pipeline.py b/sdk/python/kfp/dsl/_pipeline.py
@@ -209,13 +209,13 @@ def data_passing_method(self, value):
       ::
 
         from kfp.dsl import PipelineConf, data_passing_methods
-        from kubernetes.client.models import V1Volume, V1PersistentVolumeClaim
+        from kubernetes.client.models import V1Volume, V1PersistentVolumeClaimVolumeSource
         pipeline_conf = PipelineConf()
         pipeline_conf.data_passing_method =
         data_passing_methods.KubernetesVolume(
             volume=V1Volume(
                 name='data',
-                persistent_volume_claim=V1PersistentVolumeClaim('data-volume'),
+                persistent_volume_claim=V1PersistentVolumeClaimVolumeSource('data-volume'),
             ),
             path_prefix='artifact_data/',
         )