(serverName string, numReplicas uint32)
| 1042 | } |
| 1043 | |
| 1044 | func createTestServer(serverName string, numReplicas uint32) func(inc *IncrementalProcessor, g *WithT) { |
| 1045 | f := func(inc *IncrementalProcessor, g *WithT) { |
| 1046 | for i := uint32(0); i < numReplicas; i++ { |
| 1047 | err := inc.modelStore.AddServerReplica(&pba.AgentSubscribeRequest{ |
| 1048 | ServerName: serverName, |
| 1049 | ReplicaIdx: i, |
| 1050 | ReplicaConfig: &pba.ReplicaConfig{ |
| 1051 | InferenceSvc: fmt.Sprintf("%s.%d", serverName, i), |
| 1052 | InferenceHttpPort: 1234, |
| 1053 | }, |
| 1054 | }) |
| 1055 | g.Expect(err).To(BeNil()) |
| 1056 | } |
| 1057 | } |
| 1058 | return f |
| 1059 | } |
| 1060 | |
| 1061 | func createTestModel(modelName string, |
| 1062 | serverName string, |
no test coverage detected