client

Documentation

Index

• Constants
• func MLFrameworkProtoFromStr(framework string) proto.MLFramework
• type ApiCreateCheckpoint
• type ChangeStreamEventResponse
• type CheckpointDownloadResponse
• type ClientConfig
  • func NewClientConfig(configPath string) (*ClientConfig, error)
• type CreateModelApiResponse
• type FileUploadResponse
• type ModelBoxClient
  • func NewModelBoxClient(addr string) (*ModelBoxClient, error)
  • func (m *ModelBoxClient) ClusterMembers() (*proto.GetClusterMembersResponse, error)
  • func (m *ModelBoxClient) CreateAction(name, arch, object, params string) (*proto.CreateActionResponse, error)
  • func (m *ModelBoxClient) CreateCheckpoint(chk *ApiCreateCheckpoint) (*proto.CreateCheckpointResponse, error)
  • func (m *ModelBoxClient) CreateExperiment(name, owner, namespace, framework string) (string, error)
  • func (m *ModelBoxClient) CreateModel(name, owner, namespace, task, description string, metadata map[string]string, ...) (*CreateModelApiResponse, error)
  • func (m *ModelBoxClient) DownloadBlob(id, path string) (*CheckpointDownloadResponse, error)
  • func (m *ModelBoxClient) ListActions(objectId string) (*proto.ListActionsResponse, error)
  • func (m *ModelBoxClient) ListCheckpoints(experimentId string) (*proto.ListCheckpointsResponse, error)
  • func (m *ModelBoxClient) ListExperiments(namespace string) (*proto.ListExperimentsResponse, error)
  • func (m *ModelBoxClient) ListModels(namespace string) ([]*proto.Model, error)
  • func (m *ModelBoxClient) StremChangeEvents(namespace string, cb func(*ChangeStreamEventResponse) error) error
  • func (m *ModelBoxClient) UploadFile(path, parentId string, t artifacts.FileMIMEType) (*FileUploadResponse, error)

Constants

const (
	DEADLINE = 10 * time.Second
)

Functions

func MLFrameworkProtoFromStr

func MLFrameworkProtoFromStr(framework string) proto.MLFramework

Types

type ApiCreateCheckpoint

type ApiCreateCheckpoint struct {
	ExperimentId string
	Epoch        uint64
	Path         string
}

type ChangeStreamEventResponse

type ChangeStreamEventResponse struct {
	Event   uint8
	PayLoad *structpb.Value
}

type CheckpointDownloadResponse

type CheckpointDownloadResponse struct {
	Checksum       string
	ServerChecksum string
}

type ClientConfig

type ClientConfig struct {
	ServerAddr string `yaml:"server_addr"`
}

func NewClientConfig

func NewClientConfig(configPath string) (*ClientConfig, error)

type CreateModelApiResponse

type CreateModelApiResponse struct {
	Id string
}

type FileUploadResponse

type FileUploadResponse struct {
	Id       string
	Checksum string
}

type ModelBoxClient

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

func NewModelBoxClient

func NewModelBoxClient(addr string) (*ModelBoxClient, error)

func (*ModelBoxClient) ClusterMembers

func (m *ModelBoxClient) ClusterMembers() (*proto.GetClusterMembersResponse, error)

func (*ModelBoxClient) CreateAction

func (m *ModelBoxClient) CreateAction(name, arch, object, params string) (*proto.CreateActionResponse, error)

func (*ModelBoxClient) CreateCheckpoint

func (m *ModelBoxClient) CreateCheckpoint(chk *ApiCreateCheckpoint) (*proto.CreateCheckpointResponse, error)

func (*ModelBoxClient) CreateExperiment

func (m *ModelBoxClient) CreateExperiment(name, owner, namespace, framework string) (string, error)

func (*ModelBoxClient) CreateModel

func (m *ModelBoxClient) CreateModel(name, owner, namespace, task, description string, metadata map[string]string, files []*proto.FileMetadata) (*CreateModelApiResponse, error)

func (*ModelBoxClient) DownloadBlob

func (m *ModelBoxClient) DownloadBlob(id, path string) (*CheckpointDownloadResponse, error)

func (*ModelBoxClient) ListActions

func (m *ModelBoxClient) ListActions(objectId string) (*proto.ListActionsResponse, error)

func (*ModelBoxClient) ListCheckpoints

func (m *ModelBoxClient) ListCheckpoints(experimentId string) (*proto.ListCheckpointsResponse, error)

func (*ModelBoxClient) ListExperiments

func (m *ModelBoxClient) ListExperiments(namespace string) (*proto.ListExperimentsResponse, error)

func (*ModelBoxClient) ListModels

func (m *ModelBoxClient) ListModels(namespace string) ([]*proto.Model, error)

func (*ModelBoxClient) StremChangeEvents

func (m *ModelBoxClient) StremChangeEvents(namespace string, cb func(*ChangeStreamEventResponse) error) error

func (*ModelBoxClient) UploadFile

func (m *ModelBoxClient) UploadFile(path, parentId string, t artifacts.FileMIMEType) (*FileUploadResponse, error)