File tree Expand file tree Collapse file tree 1 file changed +10
-2
lines changed
Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Original file line number Diff line number Diff line change 66 "encoding/json"
77 "fmt"
88 "io"
9+ "math/rand"
910 "mime/multipart"
1011 "net/http"
1112 "os"
@@ -66,7 +67,7 @@ func (p *LocalDockerPlatform) Deploy(imageTag string, logWriter logger.Logger) (
6667 }
6768 */
6869
69- hostPort , err := shell .NextFreePort (5000 )
70+ hostPort , err := shell .NextFreePort (5000 + rand . Intn ( 1000 ) )
7071 if err != nil {
7172 return nil , err
7273 }
@@ -193,6 +194,12 @@ func (d *LocalDockerDeployment) RunInference(input *Example, logWriter logger.Lo
193194 if err := mwriter .Close (); err != nil {
194195 return nil , fmt .Errorf ("Failed to close form mime writer: %w" , err )
195196 }
197+
198+ _ , usedCPUSecsStart , err := d .getResourceUsage ()
199+ if err != nil {
200+ return nil , err
201+ }
202+
196203 url := fmt .Sprintf ("http://localhost:%d/infer" , d .port )
197204 req , err := http .NewRequest (http .MethodPost , url , bodyBuffer )
198205 if err != nil {
@@ -209,10 +216,11 @@ func (d *LocalDockerDeployment) RunInference(input *Example, logWriter logger.Lo
209216 }
210217 defer resp .Body .Close ()
211218
212- usedMemoryBytes , usedCPUSecs , err := d .getResourceUsage ()
219+ usedMemoryBytes , usedCPUSecsEnd , err := d .getResourceUsage ()
213220 if err != nil {
214221 return nil , err
215222 }
223+ usedCPUSecs := usedCPUSecsEnd - usedCPUSecsStart
216224
217225 if resp .StatusCode == http .StatusBadRequest {
218226 body := struct {
You can’t perform that action at this time.
0 commit comments