| 789 | } |
| 790 | |
| 791 | func (serv *OnlineServer) TrainingData(req *srv.TrainingDataRequest, stream srv.Feature_TrainingDataServer) error { |
| 792 | serv.Logger.Infow("Serving Training Data", "id", req.Id.String()) |
| 793 | client, err := serv.client.TrainingData(context.Background(), req) |
| 794 | if err != nil { |
| 795 | return fmt.Errorf("could not serve training data: %w", err) |
| 796 | } |
| 797 | for { |
| 798 | row, err := client.Recv() |
| 799 | if err != nil { |
| 800 | if err == io.EOF { |
| 801 | return nil |
| 802 | } |
| 803 | return fmt.Errorf("receive error: %w", err) |
| 804 | } |
| 805 | if err := stream.Send(row); err != nil { |
| 806 | serv.Logger.Errorw("Failed to write to stream", "Error", err) |
| 807 | return fmt.Errorf("training send row: %w", err) |
| 808 | } |
| 809 | } |
| 810 | } |
| 811 | |
| 812 | func (serv *OnlineServer) TrainingTestSplit(stream srv.Feature_TrainingTestSplitServer) error { |
| 813 | serv.Logger.Infow("Starting Training Test Split Stream") |