Improved MME support for TF

Author: ivan-khvostishkov

tests/source_dir/inference_tf_clean_model2/inference.py

@@ -48,7 +48,7 @@ def _process_output(data, context):
 
     if response_content_type == 'application/json':
         d_json = json.loads(d)
-        d[0] = d[0] + 20000  # alter prediction for model 2
+        d_json['predictions'][0] = d_json['predictions'][0] + 20000  # alter prediction for model 2
         d = json.dumps(d_json['predictions'])
 
     prediction = d

tests/source_dir/inference_tf_model2/inference.py

@@ -55,7 +55,7 @@ def _process_output(data, context):
 
     if response_content_type == 'application/json':
         d_json = json.loads(d)
-        d[0] = d[0] + 20000  # alter prediction for model 2
+        d_json['predictions'][0] = d_json['predictions'][0] + 20000  # alter prediction for model 2
         d = json.dumps(d_json['predictions'])
 
     prediction = d