Skip to content

Commit

Permalink
Simplify volume usage with containerop (#783)
Browse files Browse the repository at this point in the history
change function name to mount_pvc
  • Loading branch information
swiftdiaries authored and k8s-ci-robot committed Feb 28, 2019
1 parent 18878f1 commit 9010329
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions sdk/python/kfp/onprem.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@

def mount_pvc(pvc_name='pipeline-claim', volume_name='pipeline', volume_mount_path='/mnt/pipeline'):
"""
Modifier function to apply to a Container Op to simplify volume, volume mount addition and
enable better reuse of volumes, volume claims across container ops.
Usage:
train = train_op(...)
train.apply(mount_pvc('claim-name', 'pipeline', '/mnt/pipeline'))
"""
def _mount_pvc(task):
from kubernetes import client as k8s_client
local_pvc = k8s_client.V1PersistentVolumeClaimVolumeSource(claim_name=pvc_name)
return (
task
.add_volume(
k8s_client.V1Volume(name=volume_name, persistent_volume_claim=local_pvc)
)
.add_volume_mount(
k8s_client.V1VolumeMount(mount_path=volume_mount_path, name=volume_name)
)
)
return _mount_pvc

0 comments on commit 9010329

Please sign in to comment.