From 70fc27e037e525e4acf08417a49d2441b6ab50f3 Mon Sep 17 00:00:00 2001 From: Alexey Volkov Date: Thu, 6 Dec 2018 12:27:23 -0800 Subject: [PATCH] SDK/Components - Allow dict type spec --- sdk/python/kfp/components/_structures.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/python/kfp/components/_structures.py b/sdk/python/kfp/components/_structures.py index 052e8002bf2..16c1853209d 100644 --- a/sdk/python/kfp/components/_structures.py +++ b/sdk/python/kfp/components/_structures.py @@ -68,7 +68,7 @@ def from_struct(cls, struct:Union[Tuple[str, Mapping],Mapping[str,Mapping],str]) if 'type' in spec_dict: port_spec.type = spec_dict.pop('type') - check_instance_type(port_spec.type, [str, list]) #TODO: Check format further + check_instance_type(port_spec.type, [str, list, dict]) #TODO: Check format further if 'description' in spec_dict: port_spec.description = str(spec_dict.pop('description'))