Validate prepared sample

deliahu · deliahu · commit e8df4f59565a · 2019-08-12T16:17:02.000-07:00
diff --git a/pkg/workloads/cortex/tf_api/api.py b/pkg/workloads/cortex/tf_api/api.py
@@ -257,7 +257,7 @@ def run_predict(sample):
         )
         logger.info("pre_inference: " + util.pp_str_flat(prepared_sample))
 
-    validate_sample(sample)
+    validate_sample(prepared_sample)
 
     if util.is_resource_ref(local_cache["api"]["model"]):
         for column in local_cache["required_inputs"]:

Original file line number	Diff line number	Diff line change
`@@ -257,7 +257,7 @@ def run_predict(sample):`
`257`	`257`	`)`
`258`	`258`	`logger.info("pre_inference: " + util.pp_str_flat(prepared_sample))`
`259`	`259`
`260`		`- validate_sample(sample)`
	`260`	`+ validate_sample(prepared_sample)`
`261`	`261`
`262`	`262`	`if util.is_resource_ref(local_cache["api"]["model"]):`
`263`	`263`	`for column in local_cache["required_inputs"]:`