Use Tensorflow to infer shape for request to serving (#345)

Author: 1vn

pkg/workloads/cortex/tf_api/api.py

@@ -161,19 +161,15 @@ def create_raw_prediction_request(sample):
 
         if util.is_list(value):
             shape = [len(value)]
-            for dim in signature_def[signature_key]["inputs"][column_name]["tensorShape"]["dim"][
-                1:
-            ]:
+            for dim in signature_def[signature_key]["inputs"][column_name]["tensorShape"]["dim"]:
                 shape.append(int(dim["size"]))
         else:
             shape = [1]
-            value = [value]
+
         sig_type = signature_def[signature_key]["inputs"][column_name]["dtype"]
 
         try:
-            tensor_proto = tf.make_tensor_proto(
-                value, dtype=DTYPE_TO_TF_TYPE[sig_type], shape=shape
-            )
+            tensor_proto = tf.make_tensor_proto(value, dtype=DTYPE_TO_TF_TYPE[sig_type])
             prediction_request.inputs[column_name].CopyFrom(tensor_proto)
         except Exception as e:
             raise UserException(