kfp

package
v0.0.0-...-b1d074a Latest Latest
Warning

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

 Go to latest Published: Mar 19, 2024 License: BSD-3-Clause Imports: 50 Imported by: 0

Documentation

Overview

Package kfp is a generated GoMock package.

Package kfp is a generated GoMock package.

Index

Constants

View Source 
const (
	PushedModelArtifactType    = "PushedModel"
	ArtifactNameCustomProperty = "name"
	PushedCustomProperty       = "pushed"
	PipelineRunTypeName        = "pipeline_run"
	InvalidId                  = 0
)

Variables

This section is empty.

Functions

This section is empty.

Types

type ExperimentService 

type ExperimentService struct {
	*experiment_service.Client
}

func NewExperimentService 

func NewExperimentService(providerConfig KfpProviderConfig) (*ExperimentService, error)

func (*ExperimentService) ExperimentIdByName 

func (es *ExperimentService) ExperimentIdByName(ctx context.Context, experimentName string) (string, error)

type GrpcKfpApi 

type GrpcKfpApi struct {
	RunServiceClient go_client.RunServiceClient
	JobServiceClient go_client.JobServiceClient
}

func ConnectToKfpApi 

func ConnectToKfpApi(address string) (*GrpcKfpApi, error)

func (*GrpcKfpApi) GetResourceReferences 

func (gka *GrpcKfpApi) GetResourceReferences(ctx context.Context, runId string) (ResourceReferences, error)

func (*GrpcKfpApi) GetResourceReferencesFromDescription 

func (gka *GrpcKfpApi) GetResourceReferencesFromDescription(description string) (ResourceReferences, bool, error)

type GrpcMetadataStore 

type GrpcMetadataStore struct {
	MetadataStoreServiceClient ml_metadata.MetadataStoreServiceClient
}

func ConnectToMetadataStore 

func ConnectToMetadataStore(address string) (*GrpcMetadataStore, error)

func (*GrpcMetadataStore) GetArtifacts 

func (gms *GrpcMetadataStore) GetArtifacts(ctx context.Context, workflowName string, artifactDefs []pipelinesv1.OutputArtifact) (artifacts []common.Artifact, err error)

func (*GrpcMetadataStore) GetServingModelArtifact 

func (gms *GrpcMetadataStore) GetServingModelArtifact(ctx context.Context, workflowName string) ([]common.Artifact, error)

type KfpApi 

type KfpApi interface {
	GetResourceReferences(ctx context.Context, runId string) (ResourceReferences, error)
}

type KfpEventSourceConfig 

type KfpEventSourceConfig struct {
	KfpNamespace string `yaml:"kfpNamespace"`
}

type KfpEventingServer 

type KfpEventingServer struct {
	generic.UnimplementedEventingServer
	ProviderConfig KfpProviderConfig
	K8sClient      dynamic.Interface
	MetadataStore  MetadataStore
	KfpApi         KfpApi
	Logger         logr.Logger
}

func (*KfpEventingServer) StartEventSource 

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

type KfpProvider 

type KfpProvider struct{}

func (KfpProvider) CreateExperiment 

func (kfpp KfpProvider) CreateExperiment(ctx context.Context, providerConfig KfpProviderConfig, experimentDefinition ExperimentDefinition) (string, error)

func (KfpProvider) CreatePipeline 

func (kfpp KfpProvider) CreatePipeline(ctx context.Context, providerConfig KfpProviderConfig, pipelineDefinition PipelineDefinition, pipelineFileName string) (string, error)

func (KfpProvider) CreateRun 

func (kfpp KfpProvider) CreateRun(ctx context.Context, providerConfig KfpProviderConfig, runDefinition RunDefinition) (string, error)

func (KfpProvider) CreateRunSchedule 

func (kfpp KfpProvider) CreateRunSchedule(ctx context.Context, providerConfig KfpProviderConfig, runScheduleDefinition RunScheduleDefinition) (string, error)

func (KfpProvider) DeleteExperiment 

func (kfpp KfpProvider) DeleteExperiment(ctx context.Context, providerConfig KfpProviderConfig, id string) error

func (KfpProvider) DeletePipeline 

func (kfpp KfpProvider) DeletePipeline(ctx context.Context, providerConfig KfpProviderConfig, id string) error

func (KfpProvider) DeleteRun 

func (kfpp KfpProvider) DeleteRun(_ context.Context, _ KfpProviderConfig, _ string) error

func (KfpProvider) DeleteRunSchedule 

func (kfpp KfpProvider) DeleteRunSchedule(ctx context.Context, providerConfig KfpProviderConfig, id string) error

func (KfpProvider) EventingServer 

func (kfpp KfpProvider) EventingServer(ctx context.Context, providerConfig KfpProviderConfig) (generic.EventingServer, error)

func (KfpProvider) UpdateExperiment 

func (kfpp KfpProvider) UpdateExperiment(ctx context.Context, providerConfig KfpProviderConfig, experimentDefinition ExperimentDefinition, id string) (string, error)

func (KfpProvider) UpdatePipeline 

func (kfpp KfpProvider) UpdatePipeline(ctx context.Context, providerConfig KfpProviderConfig, pipelineDefinition PipelineDefinition, id string, pipelineFile string) (string, error)

func (KfpProvider) UpdateRunSchedule 

func (kfpp KfpProvider) UpdateRunSchedule(ctx context.Context, providerConfig KfpProviderConfig, runScheduleDefinition RunScheduleDefinition, id string) (string, error)

type KfpProviderConfig 

type KfpProviderConfig struct {
	Name                     string `yaml:"name"`
	RestKfpApiUrl            string `yaml:"restKfpApiUrl,omitempty"`
	GrpcMetadataStoreAddress string `yaml:"grpcMetadataStoreAddress,omitempty"`
	GrpcKfpApiAddress        string `yaml:"grpcKfpApiAddress,omitempty"`
}

type MetadataStore 

type MetadataStore interface {
	GetServingModelArtifact(ctx context.Context, workflowName string) ([]common.Artifact, error)
	GetArtifacts(ctx context.Context, workflowName string, artifactDefs []pipelinesv1.OutputArtifact) ([]common.Artifact, error)
}

type MockJobServiceClient 

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

MockJobServiceClient is a mock of JobServiceClient interface.

func NewMockJobServiceClient 

func NewMockJobServiceClient(ctrl *gomock.Controller) *MockJobServiceClient

NewMockJobServiceClient creates a new mock instance.

func (*MockJobServiceClient) CreateJob 

func (m *MockJobServiceClient) CreateJob(arg0 context.Context, arg1 *go_client.CreateJobRequest, arg2 ...grpc.CallOption) (*go_client.Job, error)

CreateJob mocks base method.

func (*MockJobServiceClient) DeleteJob 

func (m *MockJobServiceClient) DeleteJob(arg0 context.Context, arg1 *go_client.DeleteJobRequest, arg2 ...grpc.CallOption) (*emptypb.Empty, error)

DeleteJob mocks base method.

func (*MockJobServiceClient) DisableJob 

func (m *MockJobServiceClient) DisableJob(arg0 context.Context, arg1 *go_client.DisableJobRequest, arg2 ...grpc.CallOption) (*emptypb.Empty, error)

DisableJob mocks base method.

func (*MockJobServiceClient) EXPECT 

func (m *MockJobServiceClient) EXPECT() *MockJobServiceClientMockRecorder

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

func (*MockJobServiceClient) EnableJob 

func (m *MockJobServiceClient) EnableJob(arg0 context.Context, arg1 *go_client.EnableJobRequest, arg2 ...grpc.CallOption) (*emptypb.Empty, error)

EnableJob mocks base method.

func (*MockJobServiceClient) GetJob 

func (m *MockJobServiceClient) GetJob(arg0 context.Context, arg1 *go_client.GetJobRequest, arg2 ...grpc.CallOption) (*go_client.Job, error)

GetJob mocks base method.

func (*MockJobServiceClient) ListJobs 

func (m *MockJobServiceClient) ListJobs(arg0 context.Context, arg1 *go_client.ListJobsRequest, arg2 ...grpc.CallOption) (*go_client.ListJobsResponse, error)

ListJobs mocks base method.

type MockJobServiceClientMockRecorder 

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

MockJobServiceClientMockRecorder is the mock recorder for MockJobServiceClient.

func (*MockJobServiceClientMockRecorder) CreateJob 

func (mr *MockJobServiceClientMockRecorder) CreateJob(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

CreateJob indicates an expected call of CreateJob.

func (*MockJobServiceClientMockRecorder) DeleteJob 

func (mr *MockJobServiceClientMockRecorder) DeleteJob(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

DeleteJob indicates an expected call of DeleteJob.

func (*MockJobServiceClientMockRecorder) DisableJob 

func (mr *MockJobServiceClientMockRecorder) DisableJob(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

DisableJob indicates an expected call of DisableJob.

func (*MockJobServiceClientMockRecorder) EnableJob 

func (mr *MockJobServiceClientMockRecorder) EnableJob(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

EnableJob indicates an expected call of EnableJob.

func (*MockJobServiceClientMockRecorder) GetJob 

func (mr *MockJobServiceClientMockRecorder) GetJob(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

GetJob indicates an expected call of GetJob.

func (*MockJobServiceClientMockRecorder) ListJobs 

func (mr *MockJobServiceClientMockRecorder) ListJobs(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

ListJobs indicates an expected call of ListJobs.

type MockRunServiceClient 

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

MockRunServiceClient is a mock of RunServiceClient interface.

func NewMockRunServiceClient 

func NewMockRunServiceClient(ctrl *gomock.Controller) *MockRunServiceClient

NewMockRunServiceClient creates a new mock instance.

func (*MockRunServiceClient) ArchiveRun 

func (m *MockRunServiceClient) ArchiveRun(arg0 context.Context, arg1 *go_client.ArchiveRunRequest, arg2 ...grpc.CallOption) (*emptypb.Empty, error)

ArchiveRun mocks base method.

func (*MockRunServiceClient) CreateRun 

func (m *MockRunServiceClient) CreateRun(arg0 context.Context, arg1 *go_client.CreateRunRequest, arg2 ...grpc.CallOption) (*go_client.RunDetail, error)

CreateRun mocks base method.

func (*MockRunServiceClient) DeleteRun 

func (m *MockRunServiceClient) DeleteRun(arg0 context.Context, arg1 *go_client.DeleteRunRequest, arg2 ...grpc.CallOption) (*emptypb.Empty, error)

DeleteRun mocks base method.

func (*MockRunServiceClient) EXPECT 

func (m *MockRunServiceClient) EXPECT() *MockRunServiceClientMockRecorder

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

func (*MockRunServiceClient) GetRun 

func (m *MockRunServiceClient) GetRun(arg0 context.Context, arg1 *go_client.GetRunRequest, arg2 ...grpc.CallOption) (*go_client.RunDetail, error)

GetRun mocks base method.

func (*MockRunServiceClient) ListRuns 

func (m *MockRunServiceClient) ListRuns(arg0 context.Context, arg1 *go_client.ListRunsRequest, arg2 ...grpc.CallOption) (*go_client.ListRunsResponse, error)

ListRuns mocks base method.

func (*MockRunServiceClient) ReadArtifact 

func (m *MockRunServiceClient) ReadArtifact(arg0 context.Context, arg1 *go_client.ReadArtifactRequest, arg2 ...grpc.CallOption) (*go_client.ReadArtifactResponse, error)

ReadArtifact mocks base method.

func (*MockRunServiceClient) ReportRunMetrics 

func (m *MockRunServiceClient) ReportRunMetrics(arg0 context.Context, arg1 *go_client.ReportRunMetricsRequest, arg2 ...grpc.CallOption) (*go_client.ReportRunMetricsResponse, error)

ReportRunMetrics mocks base method.

func (*MockRunServiceClient) RetryRun 

func (m *MockRunServiceClient) RetryRun(arg0 context.Context, arg1 *go_client.RetryRunRequest, arg2 ...grpc.CallOption) (*emptypb.Empty, error)

RetryRun mocks base method.

func (*MockRunServiceClient) TerminateRun 

func (m *MockRunServiceClient) TerminateRun(arg0 context.Context, arg1 *go_client.TerminateRunRequest, arg2 ...grpc.CallOption) (*emptypb.Empty, error)

TerminateRun mocks base method.

func (*MockRunServiceClient) UnarchiveRun 

func (m *MockRunServiceClient) UnarchiveRun(arg0 context.Context, arg1 *go_client.UnarchiveRunRequest, arg2 ...grpc.CallOption) (*emptypb.Empty, error)

UnarchiveRun mocks base method.

type MockRunServiceClientMockRecorder 

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

MockRunServiceClientMockRecorder is the mock recorder for MockRunServiceClient.

func (*MockRunServiceClientMockRecorder) ArchiveRun 

func (mr *MockRunServiceClientMockRecorder) ArchiveRun(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

ArchiveRun indicates an expected call of ArchiveRun.

func (*MockRunServiceClientMockRecorder) CreateRun 

func (mr *MockRunServiceClientMockRecorder) CreateRun(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

CreateRun indicates an expected call of CreateRun.

func (*MockRunServiceClientMockRecorder) DeleteRun 

func (mr *MockRunServiceClientMockRecorder) DeleteRun(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

DeleteRun indicates an expected call of DeleteRun.

func (*MockRunServiceClientMockRecorder) GetRun 

func (mr *MockRunServiceClientMockRecorder) GetRun(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

GetRun indicates an expected call of GetRun.

func (*MockRunServiceClientMockRecorder) ListRuns 

func (mr *MockRunServiceClientMockRecorder) ListRuns(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

ListRuns indicates an expected call of ListRuns.

func (*MockRunServiceClientMockRecorder) ReadArtifact 

func (mr *MockRunServiceClientMockRecorder) ReadArtifact(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

ReadArtifact indicates an expected call of ReadArtifact.

func (*MockRunServiceClientMockRecorder) ReportRunMetrics 

func (mr *MockRunServiceClientMockRecorder) ReportRunMetrics(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

ReportRunMetrics indicates an expected call of ReportRunMetrics.

func (*MockRunServiceClientMockRecorder) RetryRun 

func (mr *MockRunServiceClientMockRecorder) RetryRun(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

RetryRun indicates an expected call of RetryRun.

func (*MockRunServiceClientMockRecorder) TerminateRun 

func (mr *MockRunServiceClientMockRecorder) TerminateRun(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

TerminateRun indicates an expected call of TerminateRun.

func (*MockRunServiceClientMockRecorder) UnarchiveRun 

func (mr *MockRunServiceClientMockRecorder) UnarchiveRun(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

UnarchiveRun indicates an expected call of UnarchiveRun.

type PipelineService 

type PipelineService struct {
	*pipeline_service.Client
}

func NewPipelineService 

func NewPipelineService(providerConfig KfpProviderConfig) (*PipelineService, error)

func (*PipelineService) PipelineIdForName 

func (ps *PipelineService) PipelineIdForName(ctx context.Context, pipelineName string) (string, error)

func (*PipelineService) PipelineVersionIdForName 

func (ps *PipelineService) PipelineVersionIdForName(ctx context.Context, versionName string, pipelineId string) (string, error)

type PipelineSpec 

type PipelineSpec struct {
	Name string `json:"name"`
}

type ResourceReferences 

type ResourceReferences struct {
	PipelineName         common.NamespacedName        `yaml:"pipelineName"`
	RunConfigurationName common.NamespacedName        `yaml:"runConfigurationName"`
	RunName              common.NamespacedName        `yaml:"runName"`
	Artifacts            []pipelinesv1.OutputArtifact `yaml:"artifacts,omitempty"`
}

Source Files

View all Source files

Directories

Path Synopsis
Package ml_metadata is a generated GoMock package.
 Package ml_metadata is a generated GoMock package.

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.