Check if model input is dict before accessing

deliahu · deliahu · commit 9e4910e1e93e · 2019-06-17T14:01:51.000-07:00
(cherry picked from commit 80e091e)
diff --git a/pkg/workloads/tf_api/api.py b/pkg/workloads/tf_api/api.py
@@ -401,7 +401,7 @@ def start(args):
 
         local_cache["required_inputs"] = tf_lib.get_base_input_columns(model["name"], ctx)
 
-        if model["input"].get("target_vocab") is not None:
+        if util.is_dict(model["input"]) and model["input"].get("target_vocab") is not None:
             local_cache["target_vocab_populated"] = ctx.populate_values(
                 model["input"]["target_vocab"], None, False
             )

Original file line number	Diff line number	Diff line change
`@@ -401,7 +401,7 @@ def start(args):`
`401`	`401`
`402`	`402`	`local_cache["required_inputs"] = tf_lib.get_base_input_columns(model["name"], ctx)`
`403`	`403`
`404`		`- if model["input"].get("target_vocab") is not None:`
	`404`	`+ if util.is_dict(model["input"]) and model["input"].get("target_vocab") is not None:`
`405`	`405`	`local_cache["target_vocab_populated"] = ctx.populate_values(`
`406`	`406`	`model["input"]["target_vocab"], None, False`
`407`	`407`	`)`