def test_proto_feedback_custom():
user_object = UserObjectLowLevel()
app = SeldonModelGRPC(user_object)
arr = np.array([1, 2])
datadef = prediction_pb2.DefaultData(
tensor=prediction_pb2.Tensor(shape=(2, 1), values=arr))
request = prediction_pb2.SeldonMessage(data=datadef)
feedback = prediction_pb2.Feedback(request=request, reward=1.0)
resp = app.SendFeedback(feedback, None)
def test_proto_feedback():
user_object = UserObject()
app = SeldonModelGRPC(user_object)
arr = np.array([1, 2])
datadef = prediction_pb2.DefaultData(
tensor=prediction_pb2.Tensor(shape=(2, 1), values=arr))
meta = prediction_pb2.Meta()
metaJson = {}
routing = {"1": 1}
metaJson["routing"] = routing
json_format.ParseDict(metaJson, meta)
request = prediction_pb2.SeldonMessage(data=datadef)
response = prediction_pb2.SeldonMessage(meta=meta, data=datadef)
feedback = prediction_pb2.Feedback(request=request,
response=response,
reward=1.0)
resp = app.SendFeedback(feedback, None)