client

Documentation

Index

• type Client
  • func MakeClient(logger *zap.Logger, schedulerUrl string) *Client
  • func (c *Client) FinishJob(jobId string) error
  • func (c *Client) SubmitInferenceTask(req []byte) ([]byte, error)
  • func (c *Client) SubmitTrainTask(req api.TrainRequest) (string, error)
  • func (c *Client) UpdateJob(task *api.TrainTask) error

Types

type Client

type Client struct {
	// contains filtered or unexported fields
}

Client gives access

func MakeClient

func MakeClient(logger *zap.Logger, schedulerUrl string) *Client

MakeClient creates a client for the scheduler

func (*Client) FinishJob

func (c *Client) FinishJob(jobId string) error

FinishJob makes the scheduler delete the job entry from the cache

func (*Client) SubmitInferenceTask

func (c *Client) SubmitInferenceTask(req []byte) ([]byte, error)

SubmitInferenceTask submits an inference task to the scheduler and returns the response from the inference task as a byte array

func (*Client) SubmitTrainTask

func (c *Client) SubmitTrainTask(req api.TrainRequest) (string, error)

SubmitTrainTask submits a training task to the scheduler

func (*Client) UpdateJob

func (c *Client) UpdateJob(task *api.TrainTask) error

UpdateJob sends a request to the scheduler to determine the new level of parallelism that should be given to a job based on metrics and previous epochs