vai

Documentation

Overview

Package vai is a generated GoMock package.

Index

• Constants
• type MockPipelineJobClient
  • func NewMockPipelineJobClient(ctrl *gomock.Controller) *MockPipelineJobClient
  • func (m *MockPipelineJobClient) EXPECT() *MockPipelineJobClientMockRecorder
  • func (m *MockPipelineJobClient) GetPipelineJob(arg0 context.Context, arg1 *aiplatformpb.GetPipelineJobRequest, ...) (*aiplatformpb.PipelineJob, error)
• type MockPipelineJobClientMockRecorder
  • func (mr *MockPipelineJobClientMockRecorder) GetPipelineJob(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call
• type PipelineJobClient
• type VAILogEntry
• type VAIProvider
  • func (vaip VAIProvider) CreateExperiment(_ context.Context, _ VAIProviderConfig, _ ExperimentDefinition) (string, error)
  • func (vaip VAIProvider) CreatePipeline(ctx context.Context, providerConfig VAIProviderConfig, ...) (string, error)
  • func (vaip VAIProvider) CreateRun(ctx context.Context, providerConfig VAIProviderConfig, ...) (string, error)
  • func (vaip VAIProvider) CreateRunSchedule(ctx context.Context, providerConfig VAIProviderConfig, ...) (string, error)
  • func (vaip VAIProvider) DeleteExperiment(_ context.Context, _ VAIProviderConfig, _ string) error
  • func (vaip VAIProvider) DeletePipeline(ctx context.Context, providerConfig VAIProviderConfig, id string) error
  • func (vaip VAIProvider) DeleteRun(_ context.Context, _ VAIProviderConfig, _ string) error
  • func (vaip VAIProvider) DeleteRunSchedule(ctx context.Context, providerConfig VAIProviderConfig, scheduleId string) error
  • func (vaip VAIProvider) EventingServer(ctx context.Context, providerConfig VAIProviderConfig) (generic.EventingServer, error)
  • func (vaip VAIProvider) UpdateExperiment(_ context.Context, _ VAIProviderConfig, _ ExperimentDefinition, _ string) (string, error)
  • func (vaip VAIProvider) UpdatePipeline(ctx context.Context, providerConfig VAIProviderConfig, ...) (string, error)
  • func (vaip VAIProvider) UpdateRunSchedule(ctx context.Context, providerConfig VAIProviderConfig, ...) (string, error)
• type VAIProviderConfig
• type VAIResource
• type VaiEventSourceConfig
• type VaiEventingServer
  • func (es *VaiEventingServer) StartEventSource(source *generic.EventSource, stream generic.Eventing_StartEventSourceServer) error

Constants

const (
	PushedModelArtifactType        = "tfx.PushedModel"
	ModelPushedMetadataProperty    = "pushed"
	ModelPushedMetadataValue       = 1
	ModelPushedDestinationProperty = "pushed_destination"
)

Types

type MockPipelineJobClient

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

MockPipelineJobClient is a mock of PipelineJobClient interface.

func NewMockPipelineJobClient

func NewMockPipelineJobClient(ctrl *gomock.Controller) *MockPipelineJobClient

NewMockPipelineJobClient creates a new mock instance.

func (*MockPipelineJobClient) EXPECT

func (m *MockPipelineJobClient) EXPECT() *MockPipelineJobClientMockRecorder

EXPECT returns an object that allows the caller to indicate expected use.

func (*MockPipelineJobClient) GetPipelineJob

func (m *MockPipelineJobClient) GetPipelineJob(arg0 context.Context, arg1 *aiplatformpb.GetPipelineJobRequest, arg2 ...gax.CallOption) (*aiplatformpb.PipelineJob, error)

GetPipelineJob mocks base method.

type MockPipelineJobClientMockRecorder

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

MockPipelineJobClientMockRecorder is the mock recorder for MockPipelineJobClient.

func (*MockPipelineJobClientMockRecorder) GetPipelineJob

func (mr *MockPipelineJobClientMockRecorder) GetPipelineJob(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

GetPipelineJob indicates an expected call of GetPipelineJob.

type PipelineJobClient

type PipelineJobClient interface {
	GetPipelineJob(ctx context.Context, req *aiplatformpb.GetPipelineJobRequest, opts ...gax.CallOption) (*aiplatformpb.PipelineJob, error)
}

type VAILogEntry

type VAILogEntry struct {
	Labels   map[string]string `json:"labels"`
	Resource VAIResource       `json:"resource"`
}

type VAIProvider

type VAIProvider struct {
}

func (VAIProvider) CreateExperiment

func (vaip VAIProvider) CreateExperiment(_ context.Context, _ VAIProviderConfig, _ ExperimentDefinition) (string, error)

func (VAIProvider) CreatePipeline

func (vaip VAIProvider) CreatePipeline(ctx context.Context, providerConfig VAIProviderConfig, pipelineDefinition PipelineDefinition, pipelineFile string) (string, error)

func (VAIProvider) CreateRun

func (vaip VAIProvider) CreateRun(ctx context.Context, providerConfig VAIProviderConfig, runDefinition RunDefinition) (string, error)

func (VAIProvider) CreateRunSchedule

func (vaip VAIProvider) CreateRunSchedule(ctx context.Context, providerConfig VAIProviderConfig, runScheduleDefinition RunScheduleDefinition) (string, error)

func (VAIProvider) DeleteExperiment

func (vaip VAIProvider) DeleteExperiment(_ context.Context, _ VAIProviderConfig, _ string) error

func (VAIProvider) DeletePipeline

func (vaip VAIProvider) DeletePipeline(ctx context.Context, providerConfig VAIProviderConfig, id string) error

func (VAIProvider) DeleteRun

func (vaip VAIProvider) DeleteRun(_ context.Context, _ VAIProviderConfig, _ string) error

func (VAIProvider) DeleteRunSchedule

func (vaip VAIProvider) DeleteRunSchedule(ctx context.Context, providerConfig VAIProviderConfig, scheduleId string) error

func (VAIProvider) EventingServer

func (vaip VAIProvider) EventingServer(ctx context.Context, providerConfig VAIProviderConfig) (generic.EventingServer, error)

func (VAIProvider) UpdateExperiment

func (vaip VAIProvider) UpdateExperiment(_ context.Context, _ VAIProviderConfig, _ ExperimentDefinition, _ string) (string, error)

func (VAIProvider) UpdatePipeline

func (vaip VAIProvider) UpdatePipeline(ctx context.Context, providerConfig VAIProviderConfig, pipelineDefinition PipelineDefinition, id string, pipelineFile string) (string, error)

func (VAIProvider) UpdateRunSchedule

func (vaip VAIProvider) UpdateRunSchedule(ctx context.Context, providerConfig VAIProviderConfig, runScheduleDefinition RunScheduleDefinition, scheduleId string) (string, error)

type VAIProviderConfig

type VAIProviderConfig struct {
	Name                                  string `yaml:"name"`
	VaiProject                            string `yaml:"vaiProject"`
	VaiLocation                           string `yaml:"vaiLocation"`
	VaiJobServiceAccount                  string `yaml:"vaiJobServiceAccount"`
	GcsEndpoint                           string `yaml:"gcsEndpoint"`
	PipelineBucket                        string `yaml:"pipelineBucket"`
	EventsourcePipelineEventsSubscription string `yaml:"eventsourcePipelineEventsSubscription"`
	MaxConcurrentRunCount                 int64  `yaml:"maxConcurrentRunCount"`
}

type VAIResource

type VAIResource struct {
	Labels map[string]string `json:"labels"`
}

type VaiEventSourceConfig

type VaiEventSourceConfig struct {
}

type VaiEventingServer

type VaiEventingServer struct {
	generic.UnimplementedEventingServer
	base.K8sApi
	ProviderConfig    VAIProviderConfig
	Logger            logr.Logger
	RunsSubscription  *pubsub.Subscription
	PipelineJobClient PipelineJobClient
}

func (*VaiEventingServer) StartEventSource

func (es *VaiEventingServer) StartEventSource(source *generic.EventSource, stream generic.Eventing_StartEventSourceServer) error