SDK/DSL: Fix bug when specifying custom resource for VolumeOp (#1595)

Closes #1593. Signed-off-by: Ilias Katsakioris <elikatsis@arrikto.com>
kubeflow · Jul 17, 2019 · 56e923f · 56e923f
1 parent d4f2b1e
commit 56e923f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk/python/kfp/dsl/_volume_op.py b/sdk/python/kfp/dsl/_volume_op.py
@@ -41,7 +41,7 @@ def __init__(self,
                  resource_name: str = None,
                  size: str = None,
                  storage_class: str = None,
-                 modes: List[str] = VOLUME_MODE_RWM,
+                 modes: List[str] = None,
                  annotations: Dict[str, str] = None,
                  data_source=None,
                  **kwargs):
@@ -112,7 +112,7 @@ def __init__(self,
             requests={"storage": size}
         )
         pvc_spec = V1PersistentVolumeClaimSpec(
-            access_modes=modes,
+            access_modes=modes or VOLUME_MODE_RWM,
             resources=requested_resources,
             storage_class_name=storage_class,
             data_source=data_source