ctx

rodrigo-castellon · May 9, 2024 · 69d1ef5 · 69d1ef5
1 parent 7565c35
commit 69d1ef5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/go/cmd/localscheduler/main.go b/go/cmd/localscheduler/main.go
@@ -57,7 +57,7 @@ func (s *server) Schedule(ctx context.Context, req *pb.ScheduleRequest) (*pb.Sch
         workerAddress := fmt.Sprintf("%s%d:%d", cfg.DNS.NodePrefix, cfg.NodeIDs.Ourself, cfg.Ports.LocalWorkerStart + worker_id)
         conn, _ := grpc.Dial(workerAddress, grpc.WithInsecure())
         workerClient := pb.NewWorkerClient(conn)
-        _, err := workerClient.Run(&pb.RunRequest{Uid: uid, Name: req.Name, Args: req.Args, Kwargs: req.Kwargs})
+        _, err := workerClient.Run(ctx, &pb.RunRequest{Uid: uid, Name: req.Name, Args: req.Args, Kwargs: req.Kwargs})
         if err != nil {
             log.Printf("cannot contact worker %d", worker_id)
         }