SDK - Components - Stop serializing string values (#2227)

This can happen with Lightweight component outputs if they've already been serialized manually.
kubeflow · Sep 26, 2019 · 7735a14 · 7735a14
1 parent 342abae
commit 7735a14
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sdk/python/kfp/components/_data_passing.py b/sdk/python/kfp/components/_data_passing.py
@@ -45,6 +45,8 @@ def _deserialize_bool(s) -> bool:
 
 
 def _serialize_json(obj) -> str:
+    if isinstance(obj, str):
+        return obj
     import json
     def default_serializer(obj):
         if hasattr(obj, 'to_struct'):
@@ -55,6 +57,8 @@ def default_serializer(obj):
 
 
 def _serialize_base64_pickle(obj) -> str:
+    if isinstance(obj, str):
+        return obj
     import base64
     import pickle
     return base64.b64encode(pickle.dumps(obj)).decode('ascii')