--- Tracked inference methods ---
(ctx context.Context, in *pb.PredictOptions, opts ...ggrpc.CallOption)
| 105 | // --- Tracked inference methods --- |
| 106 | |
| 107 | func (c *InFlightTrackingClient) Predict(ctx context.Context, in *pb.PredictOptions, opts ...ggrpc.CallOption) (*pb.Reply, error) { |
| 108 | defer c.track(ctx)() |
| 109 | reply, err := c.inner.Predict(ctx, in, opts...) |
| 110 | return reply, c.reconcile(err) |
| 111 | } |
| 112 | |
| 113 | func (c *InFlightTrackingClient) PredictStream(ctx context.Context, in *pb.PredictOptions, f func(reply *pb.Reply), opts ...ggrpc.CallOption) error { |
| 114 | defer c.track(ctx)() |