Documentation
¶
Overview ¶
Package internal is a generated GoMock package.
Index ¶
- Constants
- type EventFlow
- func (vef *EventFlow) ErrOut() <-chan error
- func (vef *EventFlow) Error(inlet streams.Inlet[error])
- func (vef *EventFlow) From(outlet streams.Outlet[StreamMessage[string]]) ...
- func (vef *EventFlow) In() chan<- StreamMessage[string]
- func (vef *EventFlow) Out() <-chan StreamMessage[*common.RunCompletionEventData]
- func (vef *EventFlow) Start()
- func (vef *EventFlow) To(inlet streams.Inlet[StreamMessage[*common.RunCompletionEventData]])
- type MockPipelineJobClient
- type MockPipelineJobClientMockRecorder
- type Parameters
- type PipelineJobClient
- type VAIProviderConfig
Constants ¶
View Source
const (
PushedModelArtifactType = "tfx.PushedModel"
ModelPushedMetadataProperty = "pushed"
ModelPushedMetadataValue = 1
ModelPushedDestinationProperty = "pushed_destination"
PipelineJobNotFinishedErr = "expected pipeline job to have finished"
)
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type EventFlow ¶
type EventFlow struct {
ProviderConfig VAIProviderConfig
PipelineJobClient PipelineJobClient
// contains filtered or unexported fields
}
func NewEventFlow ¶
func NewEventFlow(ctx context.Context, config *VAIProviderConfig, pipelineJobClient *aiplatform.PipelineClient) *EventFlow
func (*EventFlow) From ¶
func (vef *EventFlow) From(outlet streams.Outlet[StreamMessage[string]]) streams.Flow[StreamMessage[string], StreamMessage[*common.RunCompletionEventData], error]
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 Parameters ¶
type Parameters struct {
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 PipelineJobClient ¶
type PipelineJobClient interface {
GetPipelineJob(
ctx context.Context,
req *aiplatformpb.GetPipelineJobRequest,
opts ...gax.CallOption,
) (*aiplatformpb.PipelineJob, error)
}
type VAIProviderConfig ¶
type VAIProviderConfig struct {
Name string `yaml:"name"`
Parameters Parameters `yaml:"parameters"`
}
func (VAIProviderConfig) VaiEndpoint ¶
func (vaipc VAIProviderConfig) VaiEndpoint() string
Source Files
Click to show internal directories.
Click to hide internal directories.