client

package
v0.0.0-...-c4a6916 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Mar 12, 2023 License: Apache-2.0 Imports: 15 Imported by: 0

Documentation

Index

Constants

View Source 
const (
	DEADLINE = 10 * time.Second
)

Variables

This section is empty.

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)

Source Files

View all Source files

Directories

Path Synopsis

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.