Documentation ¶
Overview ¶
Package mocks is a generated GoMock package.
Index ¶
- type MockGRPCInferenceServiceClient
- func (m *MockGRPCInferenceServiceClient) CudaSharedMemoryRegister(ctx context.Context, in *inference.CudaSharedMemoryRegisterRequest, ...) (*inference.CudaSharedMemoryRegisterResponse, error)
- func (m *MockGRPCInferenceServiceClient) CudaSharedMemoryStatus(ctx context.Context, in *inference.CudaSharedMemoryStatusRequest, ...) (*inference.CudaSharedMemoryStatusResponse, error)
- func (m *MockGRPCInferenceServiceClient) CudaSharedMemoryUnregister(ctx context.Context, in *inference.CudaSharedMemoryUnregisterRequest, ...) (*inference.CudaSharedMemoryUnregisterResponse, error)
- func (m *MockGRPCInferenceServiceClient) EXPECT() *MockGRPCInferenceServiceClientMockRecorder
- func (m *MockGRPCInferenceServiceClient) LogSettings(ctx context.Context, in *inference.LogSettingsRequest, opts ...grpc.CallOption) (*inference.LogSettingsResponse, error)
- func (m *MockGRPCInferenceServiceClient) ModelConfig(ctx context.Context, in *inference.ModelConfigRequest, opts ...grpc.CallOption) (*inference.ModelConfigResponse, error)
- func (m *MockGRPCInferenceServiceClient) ModelInfer(ctx context.Context, in *inference.ModelInferRequest, opts ...grpc.CallOption) (*inference.ModelInferResponse, error)
- func (m *MockGRPCInferenceServiceClient) ModelMetadata(ctx context.Context, in *inference.ModelMetadataRequest, ...) (*inference.ModelMetadataResponse, error)
- func (m *MockGRPCInferenceServiceClient) ModelReady(ctx context.Context, in *inference.ModelReadyRequest, opts ...grpc.CallOption) (*inference.ModelReadyResponse, error)
- func (m *MockGRPCInferenceServiceClient) ModelStatistics(ctx context.Context, in *inference.ModelStatisticsRequest, ...) (*inference.ModelStatisticsResponse, error)
- func (m *MockGRPCInferenceServiceClient) ModelStreamInfer(ctx context.Context, opts ...grpc.CallOption) (inference.GRPCInferenceService_ModelStreamInferClient, error)
- func (m *MockGRPCInferenceServiceClient) RepositoryIndex(ctx context.Context, in *inference.RepositoryIndexRequest, ...) (*inference.RepositoryIndexResponse, error)
- func (m *MockGRPCInferenceServiceClient) RepositoryModelLoad(ctx context.Context, in *inference.RepositoryModelLoadRequest, ...) (*inference.RepositoryModelLoadResponse, error)
- func (m *MockGRPCInferenceServiceClient) RepositoryModelUnload(ctx context.Context, in *inference.RepositoryModelUnloadRequest, ...) (*inference.RepositoryModelUnloadResponse, error)
- func (m *MockGRPCInferenceServiceClient) ServerLive(ctx context.Context, in *inference.ServerLiveRequest, opts ...grpc.CallOption) (*inference.ServerLiveResponse, error)
- func (m *MockGRPCInferenceServiceClient) ServerMetadata(ctx context.Context, in *inference.ServerMetadataRequest, ...) (*inference.ServerMetadataResponse, error)
- func (m *MockGRPCInferenceServiceClient) ServerReady(ctx context.Context, in *inference.ServerReadyRequest, opts ...grpc.CallOption) (*inference.ServerReadyResponse, error)
- func (m *MockGRPCInferenceServiceClient) SystemSharedMemoryRegister(ctx context.Context, in *inference.SystemSharedMemoryRegisterRequest, ...) (*inference.SystemSharedMemoryRegisterResponse, error)
- func (m *MockGRPCInferenceServiceClient) SystemSharedMemoryStatus(ctx context.Context, in *inference.SystemSharedMemoryStatusRequest, ...) (*inference.SystemSharedMemoryStatusResponse, error)
- func (m *MockGRPCInferenceServiceClient) SystemSharedMemoryUnregister(ctx context.Context, in *inference.SystemSharedMemoryUnregisterRequest, ...) (*inference.SystemSharedMemoryUnregisterResponse, error)
- func (m *MockGRPCInferenceServiceClient) TraceSetting(ctx context.Context, in *inference.TraceSettingRequest, ...) (*inference.TraceSettingResponse, error)
- type MockGRPCInferenceServiceClientMockRecorder
- func (mr *MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryRegister(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryStatus(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryUnregister(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) LogSettings(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelConfig(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelInfer(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelMetadata(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelReady(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelStatistics(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelStreamInfer(ctx any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) RepositoryIndex(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) RepositoryModelLoad(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) RepositoryModelUnload(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) ServerLive(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) ServerMetadata(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) ServerReady(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryRegister(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryStatus(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryUnregister(ctx, in any, opts ...any) *gomock.Call
- func (mr *MockGRPCInferenceServiceClientMockRecorder) TraceSetting(ctx, in any, opts ...any) *gomock.Call
- type MockGRPCInferenceServiceServer
- func (m *MockGRPCInferenceServiceServer) CudaSharedMemoryRegister(arg0 context.Context, arg1 *inference.CudaSharedMemoryRegisterRequest) (*inference.CudaSharedMemoryRegisterResponse, error)
- func (m *MockGRPCInferenceServiceServer) CudaSharedMemoryStatus(arg0 context.Context, arg1 *inference.CudaSharedMemoryStatusRequest) (*inference.CudaSharedMemoryStatusResponse, error)
- func (m *MockGRPCInferenceServiceServer) CudaSharedMemoryUnregister(arg0 context.Context, arg1 *inference.CudaSharedMemoryUnregisterRequest) (*inference.CudaSharedMemoryUnregisterResponse, error)
- func (m *MockGRPCInferenceServiceServer) EXPECT() *MockGRPCInferenceServiceServerMockRecorder
- func (m *MockGRPCInferenceServiceServer) LogSettings(arg0 context.Context, arg1 *inference.LogSettingsRequest) (*inference.LogSettingsResponse, error)
- func (m *MockGRPCInferenceServiceServer) ModelConfig(arg0 context.Context, arg1 *inference.ModelConfigRequest) (*inference.ModelConfigResponse, error)
- func (m *MockGRPCInferenceServiceServer) ModelInfer(arg0 context.Context, arg1 *inference.ModelInferRequest) (*inference.ModelInferResponse, error)
- func (m *MockGRPCInferenceServiceServer) ModelMetadata(arg0 context.Context, arg1 *inference.ModelMetadataRequest) (*inference.ModelMetadataResponse, error)
- func (m *MockGRPCInferenceServiceServer) ModelReady(arg0 context.Context, arg1 *inference.ModelReadyRequest) (*inference.ModelReadyResponse, error)
- func (m *MockGRPCInferenceServiceServer) ModelStatistics(arg0 context.Context, arg1 *inference.ModelStatisticsRequest) (*inference.ModelStatisticsResponse, error)
- func (m *MockGRPCInferenceServiceServer) ModelStreamInfer(arg0 inference.GRPCInferenceService_ModelStreamInferServer) error
- func (m *MockGRPCInferenceServiceServer) RepositoryIndex(arg0 context.Context, arg1 *inference.RepositoryIndexRequest) (*inference.RepositoryIndexResponse, error)
- func (m *MockGRPCInferenceServiceServer) RepositoryModelLoad(arg0 context.Context, arg1 *inference.RepositoryModelLoadRequest) (*inference.RepositoryModelLoadResponse, error)
- func (m *MockGRPCInferenceServiceServer) RepositoryModelUnload(arg0 context.Context, arg1 *inference.RepositoryModelUnloadRequest) (*inference.RepositoryModelUnloadResponse, error)
- func (m *MockGRPCInferenceServiceServer) ServerLive(arg0 context.Context, arg1 *inference.ServerLiveRequest) (*inference.ServerLiveResponse, error)
- func (m *MockGRPCInferenceServiceServer) ServerMetadata(arg0 context.Context, arg1 *inference.ServerMetadataRequest) (*inference.ServerMetadataResponse, error)
- func (m *MockGRPCInferenceServiceServer) ServerReady(arg0 context.Context, arg1 *inference.ServerReadyRequest) (*inference.ServerReadyResponse, error)
- func (m *MockGRPCInferenceServiceServer) SystemSharedMemoryRegister(arg0 context.Context, arg1 *inference.SystemSharedMemoryRegisterRequest) (*inference.SystemSharedMemoryRegisterResponse, error)
- func (m *MockGRPCInferenceServiceServer) SystemSharedMemoryStatus(arg0 context.Context, arg1 *inference.SystemSharedMemoryStatusRequest) (*inference.SystemSharedMemoryStatusResponse, error)
- func (m *MockGRPCInferenceServiceServer) SystemSharedMemoryUnregister(arg0 context.Context, arg1 *inference.SystemSharedMemoryUnregisterRequest) (*inference.SystemSharedMemoryUnregisterResponse, error)
- func (m *MockGRPCInferenceServiceServer) TraceSetting(arg0 context.Context, arg1 *inference.TraceSettingRequest) (*inference.TraceSettingResponse, error)
- type MockGRPCInferenceServiceServerMockRecorder
- func (mr *MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryRegister(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryStatus(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryUnregister(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) LogSettings(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelConfig(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelInfer(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelMetadata(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelReady(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelStatistics(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelStreamInfer(arg0 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) RepositoryIndex(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) RepositoryModelLoad(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) RepositoryModelUnload(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) ServerLive(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) ServerMetadata(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) ServerReady(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryRegister(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryStatus(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryUnregister(arg0, arg1 any) *gomock.Call
- func (mr *MockGRPCInferenceServiceServerMockRecorder) TraceSetting(arg0, arg1 any) *gomock.Call
- type MockGRPCInferenceService_ModelStreamInferClient
- func (m *MockGRPCInferenceService_ModelStreamInferClient) CloseSend() error
- func (m *MockGRPCInferenceService_ModelStreamInferClient) Context() context.Context
- func (m *MockGRPCInferenceService_ModelStreamInferClient) EXPECT() *MockGRPCInferenceService_ModelStreamInferClientMockRecorder
- func (m *MockGRPCInferenceService_ModelStreamInferClient) Header() (metadata.MD, error)
- func (m *MockGRPCInferenceService_ModelStreamInferClient) Recv() (*inference.ModelStreamInferResponse, error)
- func (m_2 *MockGRPCInferenceService_ModelStreamInferClient) RecvMsg(m any) error
- func (m *MockGRPCInferenceService_ModelStreamInferClient) Send(arg0 *inference.ModelInferRequest) error
- func (m_2 *MockGRPCInferenceService_ModelStreamInferClient) SendMsg(m any) error
- func (m *MockGRPCInferenceService_ModelStreamInferClient) Trailer() metadata.MD
- type MockGRPCInferenceService_ModelStreamInferClientMockRecorder
- func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) CloseSend() *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Context() *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Header() *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Recv() *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) RecvMsg(m any) *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Send(arg0 any) *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) SendMsg(m any) *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Trailer() *gomock.Call
- type MockGRPCInferenceService_ModelStreamInferServer
- func (m *MockGRPCInferenceService_ModelStreamInferServer) Context() context.Context
- func (m *MockGRPCInferenceService_ModelStreamInferServer) EXPECT() *MockGRPCInferenceService_ModelStreamInferServerMockRecorder
- func (m *MockGRPCInferenceService_ModelStreamInferServer) Recv() (*inference.ModelInferRequest, error)
- func (m_2 *MockGRPCInferenceService_ModelStreamInferServer) RecvMsg(m any) error
- func (m *MockGRPCInferenceService_ModelStreamInferServer) Send(arg0 *inference.ModelStreamInferResponse) error
- func (m *MockGRPCInferenceService_ModelStreamInferServer) SendHeader(arg0 metadata.MD) error
- func (m_2 *MockGRPCInferenceService_ModelStreamInferServer) SendMsg(m any) error
- func (m *MockGRPCInferenceService_ModelStreamInferServer) SetHeader(arg0 metadata.MD) error
- func (m *MockGRPCInferenceService_ModelStreamInferServer) SetTrailer(arg0 metadata.MD)
- type MockGRPCInferenceService_ModelStreamInferServerMockRecorder
- func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Context() *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Recv() *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) RecvMsg(m any) *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Send(arg0 any) *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SendHeader(arg0 any) *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SendMsg(m any) *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SetHeader(arg0 any) *gomock.Call
- func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SetTrailer(arg0 any) *gomock.Call
- type MockUnsafeGRPCInferenceServiceServer
- type MockUnsafeGRPCInferenceServiceServerMockRecorder
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type MockGRPCInferenceServiceClient ¶
type MockGRPCInferenceServiceClient struct {
// contains filtered or unexported fields
}
MockGRPCInferenceServiceClient is a mock of GRPCInferenceServiceClient interface.
func NewMockGRPCInferenceServiceClient ¶
func NewMockGRPCInferenceServiceClient(ctrl *gomock.Controller) *MockGRPCInferenceServiceClient
NewMockGRPCInferenceServiceClient creates a new mock instance.
func (*MockGRPCInferenceServiceClient) CudaSharedMemoryRegister ¶
func (m *MockGRPCInferenceServiceClient) CudaSharedMemoryRegister(ctx context.Context, in *inference.CudaSharedMemoryRegisterRequest, opts ...grpc.CallOption) (*inference.CudaSharedMemoryRegisterResponse, error)
CudaSharedMemoryRegister mocks base method.
func (*MockGRPCInferenceServiceClient) CudaSharedMemoryStatus ¶
func (m *MockGRPCInferenceServiceClient) CudaSharedMemoryStatus(ctx context.Context, in *inference.CudaSharedMemoryStatusRequest, opts ...grpc.CallOption) (*inference.CudaSharedMemoryStatusResponse, error)
CudaSharedMemoryStatus mocks base method.
func (*MockGRPCInferenceServiceClient) CudaSharedMemoryUnregister ¶
func (m *MockGRPCInferenceServiceClient) CudaSharedMemoryUnregister(ctx context.Context, in *inference.CudaSharedMemoryUnregisterRequest, opts ...grpc.CallOption) (*inference.CudaSharedMemoryUnregisterResponse, error)
CudaSharedMemoryUnregister mocks base method.
func (*MockGRPCInferenceServiceClient) EXPECT ¶
func (m *MockGRPCInferenceServiceClient) EXPECT() *MockGRPCInferenceServiceClientMockRecorder
EXPECT returns an object that allows the caller to indicate expected use.
func (*MockGRPCInferenceServiceClient) LogSettings ¶
func (m *MockGRPCInferenceServiceClient) LogSettings(ctx context.Context, in *inference.LogSettingsRequest, opts ...grpc.CallOption) (*inference.LogSettingsResponse, error)
LogSettings mocks base method.
func (*MockGRPCInferenceServiceClient) ModelConfig ¶
func (m *MockGRPCInferenceServiceClient) ModelConfig(ctx context.Context, in *inference.ModelConfigRequest, opts ...grpc.CallOption) (*inference.ModelConfigResponse, error)
ModelConfig mocks base method.
func (*MockGRPCInferenceServiceClient) ModelInfer ¶
func (m *MockGRPCInferenceServiceClient) ModelInfer(ctx context.Context, in *inference.ModelInferRequest, opts ...grpc.CallOption) (*inference.ModelInferResponse, error)
ModelInfer mocks base method.
func (*MockGRPCInferenceServiceClient) ModelMetadata ¶
func (m *MockGRPCInferenceServiceClient) ModelMetadata(ctx context.Context, in *inference.ModelMetadataRequest, opts ...grpc.CallOption) (*inference.ModelMetadataResponse, error)
ModelMetadata mocks base method.
func (*MockGRPCInferenceServiceClient) ModelReady ¶
func (m *MockGRPCInferenceServiceClient) ModelReady(ctx context.Context, in *inference.ModelReadyRequest, opts ...grpc.CallOption) (*inference.ModelReadyResponse, error)
ModelReady mocks base method.
func (*MockGRPCInferenceServiceClient) ModelStatistics ¶
func (m *MockGRPCInferenceServiceClient) ModelStatistics(ctx context.Context, in *inference.ModelStatisticsRequest, opts ...grpc.CallOption) (*inference.ModelStatisticsResponse, error)
ModelStatistics mocks base method.
func (*MockGRPCInferenceServiceClient) ModelStreamInfer ¶
func (m *MockGRPCInferenceServiceClient) ModelStreamInfer(ctx context.Context, opts ...grpc.CallOption) (inference.GRPCInferenceService_ModelStreamInferClient, error)
ModelStreamInfer mocks base method.
func (*MockGRPCInferenceServiceClient) RepositoryIndex ¶
func (m *MockGRPCInferenceServiceClient) RepositoryIndex(ctx context.Context, in *inference.RepositoryIndexRequest, opts ...grpc.CallOption) (*inference.RepositoryIndexResponse, error)
RepositoryIndex mocks base method.
func (*MockGRPCInferenceServiceClient) RepositoryModelLoad ¶
func (m *MockGRPCInferenceServiceClient) RepositoryModelLoad(ctx context.Context, in *inference.RepositoryModelLoadRequest, opts ...grpc.CallOption) (*inference.RepositoryModelLoadResponse, error)
RepositoryModelLoad mocks base method.
func (*MockGRPCInferenceServiceClient) RepositoryModelUnload ¶
func (m *MockGRPCInferenceServiceClient) RepositoryModelUnload(ctx context.Context, in *inference.RepositoryModelUnloadRequest, opts ...grpc.CallOption) (*inference.RepositoryModelUnloadResponse, error)
RepositoryModelUnload mocks base method.
func (*MockGRPCInferenceServiceClient) ServerLive ¶
func (m *MockGRPCInferenceServiceClient) ServerLive(ctx context.Context, in *inference.ServerLiveRequest, opts ...grpc.CallOption) (*inference.ServerLiveResponse, error)
ServerLive mocks base method.
func (*MockGRPCInferenceServiceClient) ServerMetadata ¶
func (m *MockGRPCInferenceServiceClient) ServerMetadata(ctx context.Context, in *inference.ServerMetadataRequest, opts ...grpc.CallOption) (*inference.ServerMetadataResponse, error)
ServerMetadata mocks base method.
func (*MockGRPCInferenceServiceClient) ServerReady ¶
func (m *MockGRPCInferenceServiceClient) ServerReady(ctx context.Context, in *inference.ServerReadyRequest, opts ...grpc.CallOption) (*inference.ServerReadyResponse, error)
ServerReady mocks base method.
func (*MockGRPCInferenceServiceClient) SystemSharedMemoryRegister ¶
func (m *MockGRPCInferenceServiceClient) SystemSharedMemoryRegister(ctx context.Context, in *inference.SystemSharedMemoryRegisterRequest, opts ...grpc.CallOption) (*inference.SystemSharedMemoryRegisterResponse, error)
SystemSharedMemoryRegister mocks base method.
func (*MockGRPCInferenceServiceClient) SystemSharedMemoryStatus ¶
func (m *MockGRPCInferenceServiceClient) SystemSharedMemoryStatus(ctx context.Context, in *inference.SystemSharedMemoryStatusRequest, opts ...grpc.CallOption) (*inference.SystemSharedMemoryStatusResponse, error)
SystemSharedMemoryStatus mocks base method.
func (*MockGRPCInferenceServiceClient) SystemSharedMemoryUnregister ¶
func (m *MockGRPCInferenceServiceClient) SystemSharedMemoryUnregister(ctx context.Context, in *inference.SystemSharedMemoryUnregisterRequest, opts ...grpc.CallOption) (*inference.SystemSharedMemoryUnregisterResponse, error)
SystemSharedMemoryUnregister mocks base method.
func (*MockGRPCInferenceServiceClient) TraceSetting ¶
func (m *MockGRPCInferenceServiceClient) TraceSetting(ctx context.Context, in *inference.TraceSettingRequest, opts ...grpc.CallOption) (*inference.TraceSettingResponse, error)
TraceSetting mocks base method.
type MockGRPCInferenceServiceClientMockRecorder ¶
type MockGRPCInferenceServiceClientMockRecorder struct {
// contains filtered or unexported fields
}
MockGRPCInferenceServiceClientMockRecorder is the mock recorder for MockGRPCInferenceServiceClient.
func (*MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryRegister ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryRegister(ctx, in any, opts ...any) *gomock.Call
CudaSharedMemoryRegister indicates an expected call of CudaSharedMemoryRegister.
func (*MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryStatus ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryStatus(ctx, in any, opts ...any) *gomock.Call
CudaSharedMemoryStatus indicates an expected call of CudaSharedMemoryStatus.
func (*MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryUnregister ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryUnregister(ctx, in any, opts ...any) *gomock.Call
CudaSharedMemoryUnregister indicates an expected call of CudaSharedMemoryUnregister.
func (*MockGRPCInferenceServiceClientMockRecorder) LogSettings ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) LogSettings(ctx, in any, opts ...any) *gomock.Call
LogSettings indicates an expected call of LogSettings.
func (*MockGRPCInferenceServiceClientMockRecorder) ModelConfig ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelConfig(ctx, in any, opts ...any) *gomock.Call
ModelConfig indicates an expected call of ModelConfig.
func (*MockGRPCInferenceServiceClientMockRecorder) ModelInfer ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelInfer(ctx, in any, opts ...any) *gomock.Call
ModelInfer indicates an expected call of ModelInfer.
func (*MockGRPCInferenceServiceClientMockRecorder) ModelMetadata ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelMetadata(ctx, in any, opts ...any) *gomock.Call
ModelMetadata indicates an expected call of ModelMetadata.
func (*MockGRPCInferenceServiceClientMockRecorder) ModelReady ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelReady(ctx, in any, opts ...any) *gomock.Call
ModelReady indicates an expected call of ModelReady.
func (*MockGRPCInferenceServiceClientMockRecorder) ModelStatistics ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelStatistics(ctx, in any, opts ...any) *gomock.Call
ModelStatistics indicates an expected call of ModelStatistics.
func (*MockGRPCInferenceServiceClientMockRecorder) ModelStreamInfer ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelStreamInfer(ctx any, opts ...any) *gomock.Call
ModelStreamInfer indicates an expected call of ModelStreamInfer.
func (*MockGRPCInferenceServiceClientMockRecorder) RepositoryIndex ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) RepositoryIndex(ctx, in any, opts ...any) *gomock.Call
RepositoryIndex indicates an expected call of RepositoryIndex.
func (*MockGRPCInferenceServiceClientMockRecorder) RepositoryModelLoad ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) RepositoryModelLoad(ctx, in any, opts ...any) *gomock.Call
RepositoryModelLoad indicates an expected call of RepositoryModelLoad.
func (*MockGRPCInferenceServiceClientMockRecorder) RepositoryModelUnload ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) RepositoryModelUnload(ctx, in any, opts ...any) *gomock.Call
RepositoryModelUnload indicates an expected call of RepositoryModelUnload.
func (*MockGRPCInferenceServiceClientMockRecorder) ServerLive ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) ServerLive(ctx, in any, opts ...any) *gomock.Call
ServerLive indicates an expected call of ServerLive.
func (*MockGRPCInferenceServiceClientMockRecorder) ServerMetadata ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) ServerMetadata(ctx, in any, opts ...any) *gomock.Call
ServerMetadata indicates an expected call of ServerMetadata.
func (*MockGRPCInferenceServiceClientMockRecorder) ServerReady ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) ServerReady(ctx, in any, opts ...any) *gomock.Call
ServerReady indicates an expected call of ServerReady.
func (*MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryRegister ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryRegister(ctx, in any, opts ...any) *gomock.Call
SystemSharedMemoryRegister indicates an expected call of SystemSharedMemoryRegister.
func (*MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryStatus ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryStatus(ctx, in any, opts ...any) *gomock.Call
SystemSharedMemoryStatus indicates an expected call of SystemSharedMemoryStatus.
func (*MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryUnregister ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryUnregister(ctx, in any, opts ...any) *gomock.Call
SystemSharedMemoryUnregister indicates an expected call of SystemSharedMemoryUnregister.
func (*MockGRPCInferenceServiceClientMockRecorder) TraceSetting ¶
func (mr *MockGRPCInferenceServiceClientMockRecorder) TraceSetting(ctx, in any, opts ...any) *gomock.Call
TraceSetting indicates an expected call of TraceSetting.
type MockGRPCInferenceServiceServer ¶
type MockGRPCInferenceServiceServer struct {
// contains filtered or unexported fields
}
MockGRPCInferenceServiceServer is a mock of GRPCInferenceServiceServer interface.
func NewMockGRPCInferenceServiceServer ¶
func NewMockGRPCInferenceServiceServer(ctrl *gomock.Controller) *MockGRPCInferenceServiceServer
NewMockGRPCInferenceServiceServer creates a new mock instance.
func (*MockGRPCInferenceServiceServer) CudaSharedMemoryRegister ¶
func (m *MockGRPCInferenceServiceServer) CudaSharedMemoryRegister(arg0 context.Context, arg1 *inference.CudaSharedMemoryRegisterRequest) (*inference.CudaSharedMemoryRegisterResponse, error)
CudaSharedMemoryRegister mocks base method.
func (*MockGRPCInferenceServiceServer) CudaSharedMemoryStatus ¶
func (m *MockGRPCInferenceServiceServer) CudaSharedMemoryStatus(arg0 context.Context, arg1 *inference.CudaSharedMemoryStatusRequest) (*inference.CudaSharedMemoryStatusResponse, error)
CudaSharedMemoryStatus mocks base method.
func (*MockGRPCInferenceServiceServer) CudaSharedMemoryUnregister ¶
func (m *MockGRPCInferenceServiceServer) CudaSharedMemoryUnregister(arg0 context.Context, arg1 *inference.CudaSharedMemoryUnregisterRequest) (*inference.CudaSharedMemoryUnregisterResponse, error)
CudaSharedMemoryUnregister mocks base method.
func (*MockGRPCInferenceServiceServer) EXPECT ¶
func (m *MockGRPCInferenceServiceServer) EXPECT() *MockGRPCInferenceServiceServerMockRecorder
EXPECT returns an object that allows the caller to indicate expected use.
func (*MockGRPCInferenceServiceServer) LogSettings ¶
func (m *MockGRPCInferenceServiceServer) LogSettings(arg0 context.Context, arg1 *inference.LogSettingsRequest) (*inference.LogSettingsResponse, error)
LogSettings mocks base method.
func (*MockGRPCInferenceServiceServer) ModelConfig ¶
func (m *MockGRPCInferenceServiceServer) ModelConfig(arg0 context.Context, arg1 *inference.ModelConfigRequest) (*inference.ModelConfigResponse, error)
ModelConfig mocks base method.
func (*MockGRPCInferenceServiceServer) ModelInfer ¶
func (m *MockGRPCInferenceServiceServer) ModelInfer(arg0 context.Context, arg1 *inference.ModelInferRequest) (*inference.ModelInferResponse, error)
ModelInfer mocks base method.
func (*MockGRPCInferenceServiceServer) ModelMetadata ¶
func (m *MockGRPCInferenceServiceServer) ModelMetadata(arg0 context.Context, arg1 *inference.ModelMetadataRequest) (*inference.ModelMetadataResponse, error)
ModelMetadata mocks base method.
func (*MockGRPCInferenceServiceServer) ModelReady ¶
func (m *MockGRPCInferenceServiceServer) ModelReady(arg0 context.Context, arg1 *inference.ModelReadyRequest) (*inference.ModelReadyResponse, error)
ModelReady mocks base method.
func (*MockGRPCInferenceServiceServer) ModelStatistics ¶
func (m *MockGRPCInferenceServiceServer) ModelStatistics(arg0 context.Context, arg1 *inference.ModelStatisticsRequest) (*inference.ModelStatisticsResponse, error)
ModelStatistics mocks base method.
func (*MockGRPCInferenceServiceServer) ModelStreamInfer ¶
func (m *MockGRPCInferenceServiceServer) ModelStreamInfer(arg0 inference.GRPCInferenceService_ModelStreamInferServer) error
ModelStreamInfer mocks base method.
func (*MockGRPCInferenceServiceServer) RepositoryIndex ¶
func (m *MockGRPCInferenceServiceServer) RepositoryIndex(arg0 context.Context, arg1 *inference.RepositoryIndexRequest) (*inference.RepositoryIndexResponse, error)
RepositoryIndex mocks base method.
func (*MockGRPCInferenceServiceServer) RepositoryModelLoad ¶
func (m *MockGRPCInferenceServiceServer) RepositoryModelLoad(arg0 context.Context, arg1 *inference.RepositoryModelLoadRequest) (*inference.RepositoryModelLoadResponse, error)
RepositoryModelLoad mocks base method.
func (*MockGRPCInferenceServiceServer) RepositoryModelUnload ¶
func (m *MockGRPCInferenceServiceServer) RepositoryModelUnload(arg0 context.Context, arg1 *inference.RepositoryModelUnloadRequest) (*inference.RepositoryModelUnloadResponse, error)
RepositoryModelUnload mocks base method.
func (*MockGRPCInferenceServiceServer) ServerLive ¶
func (m *MockGRPCInferenceServiceServer) ServerLive(arg0 context.Context, arg1 *inference.ServerLiveRequest) (*inference.ServerLiveResponse, error)
ServerLive mocks base method.
func (*MockGRPCInferenceServiceServer) ServerMetadata ¶
func (m *MockGRPCInferenceServiceServer) ServerMetadata(arg0 context.Context, arg1 *inference.ServerMetadataRequest) (*inference.ServerMetadataResponse, error)
ServerMetadata mocks base method.
func (*MockGRPCInferenceServiceServer) ServerReady ¶
func (m *MockGRPCInferenceServiceServer) ServerReady(arg0 context.Context, arg1 *inference.ServerReadyRequest) (*inference.ServerReadyResponse, error)
ServerReady mocks base method.
func (*MockGRPCInferenceServiceServer) SystemSharedMemoryRegister ¶
func (m *MockGRPCInferenceServiceServer) SystemSharedMemoryRegister(arg0 context.Context, arg1 *inference.SystemSharedMemoryRegisterRequest) (*inference.SystemSharedMemoryRegisterResponse, error)
SystemSharedMemoryRegister mocks base method.
func (*MockGRPCInferenceServiceServer) SystemSharedMemoryStatus ¶
func (m *MockGRPCInferenceServiceServer) SystemSharedMemoryStatus(arg0 context.Context, arg1 *inference.SystemSharedMemoryStatusRequest) (*inference.SystemSharedMemoryStatusResponse, error)
SystemSharedMemoryStatus mocks base method.
func (*MockGRPCInferenceServiceServer) SystemSharedMemoryUnregister ¶
func (m *MockGRPCInferenceServiceServer) SystemSharedMemoryUnregister(arg0 context.Context, arg1 *inference.SystemSharedMemoryUnregisterRequest) (*inference.SystemSharedMemoryUnregisterResponse, error)
SystemSharedMemoryUnregister mocks base method.
func (*MockGRPCInferenceServiceServer) TraceSetting ¶
func (m *MockGRPCInferenceServiceServer) TraceSetting(arg0 context.Context, arg1 *inference.TraceSettingRequest) (*inference.TraceSettingResponse, error)
TraceSetting mocks base method.
type MockGRPCInferenceServiceServerMockRecorder ¶
type MockGRPCInferenceServiceServerMockRecorder struct {
// contains filtered or unexported fields
}
MockGRPCInferenceServiceServerMockRecorder is the mock recorder for MockGRPCInferenceServiceServer.
func (*MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryRegister ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryRegister(arg0, arg1 any) *gomock.Call
CudaSharedMemoryRegister indicates an expected call of CudaSharedMemoryRegister.
func (*MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryStatus ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryStatus(arg0, arg1 any) *gomock.Call
CudaSharedMemoryStatus indicates an expected call of CudaSharedMemoryStatus.
func (*MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryUnregister ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryUnregister(arg0, arg1 any) *gomock.Call
CudaSharedMemoryUnregister indicates an expected call of CudaSharedMemoryUnregister.
func (*MockGRPCInferenceServiceServerMockRecorder) LogSettings ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) LogSettings(arg0, arg1 any) *gomock.Call
LogSettings indicates an expected call of LogSettings.
func (*MockGRPCInferenceServiceServerMockRecorder) ModelConfig ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelConfig(arg0, arg1 any) *gomock.Call
ModelConfig indicates an expected call of ModelConfig.
func (*MockGRPCInferenceServiceServerMockRecorder) ModelInfer ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelInfer(arg0, arg1 any) *gomock.Call
ModelInfer indicates an expected call of ModelInfer.
func (*MockGRPCInferenceServiceServerMockRecorder) ModelMetadata ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelMetadata(arg0, arg1 any) *gomock.Call
ModelMetadata indicates an expected call of ModelMetadata.
func (*MockGRPCInferenceServiceServerMockRecorder) ModelReady ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelReady(arg0, arg1 any) *gomock.Call
ModelReady indicates an expected call of ModelReady.
func (*MockGRPCInferenceServiceServerMockRecorder) ModelStatistics ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelStatistics(arg0, arg1 any) *gomock.Call
ModelStatistics indicates an expected call of ModelStatistics.
func (*MockGRPCInferenceServiceServerMockRecorder) ModelStreamInfer ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelStreamInfer(arg0 any) *gomock.Call
ModelStreamInfer indicates an expected call of ModelStreamInfer.
func (*MockGRPCInferenceServiceServerMockRecorder) RepositoryIndex ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) RepositoryIndex(arg0, arg1 any) *gomock.Call
RepositoryIndex indicates an expected call of RepositoryIndex.
func (*MockGRPCInferenceServiceServerMockRecorder) RepositoryModelLoad ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) RepositoryModelLoad(arg0, arg1 any) *gomock.Call
RepositoryModelLoad indicates an expected call of RepositoryModelLoad.
func (*MockGRPCInferenceServiceServerMockRecorder) RepositoryModelUnload ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) RepositoryModelUnload(arg0, arg1 any) *gomock.Call
RepositoryModelUnload indicates an expected call of RepositoryModelUnload.
func (*MockGRPCInferenceServiceServerMockRecorder) ServerLive ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) ServerLive(arg0, arg1 any) *gomock.Call
ServerLive indicates an expected call of ServerLive.
func (*MockGRPCInferenceServiceServerMockRecorder) ServerMetadata ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) ServerMetadata(arg0, arg1 any) *gomock.Call
ServerMetadata indicates an expected call of ServerMetadata.
func (*MockGRPCInferenceServiceServerMockRecorder) ServerReady ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) ServerReady(arg0, arg1 any) *gomock.Call
ServerReady indicates an expected call of ServerReady.
func (*MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryRegister ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryRegister(arg0, arg1 any) *gomock.Call
SystemSharedMemoryRegister indicates an expected call of SystemSharedMemoryRegister.
func (*MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryStatus ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryStatus(arg0, arg1 any) *gomock.Call
SystemSharedMemoryStatus indicates an expected call of SystemSharedMemoryStatus.
func (*MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryUnregister ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryUnregister(arg0, arg1 any) *gomock.Call
SystemSharedMemoryUnregister indicates an expected call of SystemSharedMemoryUnregister.
func (*MockGRPCInferenceServiceServerMockRecorder) TraceSetting ¶
func (mr *MockGRPCInferenceServiceServerMockRecorder) TraceSetting(arg0, arg1 any) *gomock.Call
TraceSetting indicates an expected call of TraceSetting.
type MockGRPCInferenceService_ModelStreamInferClient ¶
type MockGRPCInferenceService_ModelStreamInferClient struct {
// contains filtered or unexported fields
}
MockGRPCInferenceService_ModelStreamInferClient is a mock of GRPCInferenceService_ModelStreamInferClient interface.
func NewMockGRPCInferenceService_ModelStreamInferClient ¶
func NewMockGRPCInferenceService_ModelStreamInferClient(ctrl *gomock.Controller) *MockGRPCInferenceService_ModelStreamInferClient
NewMockGRPCInferenceService_ModelStreamInferClient creates a new mock instance.
func (*MockGRPCInferenceService_ModelStreamInferClient) CloseSend ¶
func (m *MockGRPCInferenceService_ModelStreamInferClient) CloseSend() error
CloseSend mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferClient) Context ¶
func (m *MockGRPCInferenceService_ModelStreamInferClient) Context() context.Context
Context mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferClient) EXPECT ¶
func (m *MockGRPCInferenceService_ModelStreamInferClient) EXPECT() *MockGRPCInferenceService_ModelStreamInferClientMockRecorder
EXPECT returns an object that allows the caller to indicate expected use.
func (*MockGRPCInferenceService_ModelStreamInferClient) Header ¶
func (m *MockGRPCInferenceService_ModelStreamInferClient) Header() (metadata.MD, error)
Header mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferClient) Recv ¶
func (m *MockGRPCInferenceService_ModelStreamInferClient) Recv() (*inference.ModelStreamInferResponse, error)
Recv mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferClient) RecvMsg ¶
func (m_2 *MockGRPCInferenceService_ModelStreamInferClient) RecvMsg(m any) error
RecvMsg mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferClient) Send ¶
func (m *MockGRPCInferenceService_ModelStreamInferClient) Send(arg0 *inference.ModelInferRequest) error
Send mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferClient) SendMsg ¶
func (m_2 *MockGRPCInferenceService_ModelStreamInferClient) SendMsg(m any) error
SendMsg mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferClient) Trailer ¶
func (m *MockGRPCInferenceService_ModelStreamInferClient) Trailer() metadata.MD
Trailer mocks base method.
type MockGRPCInferenceService_ModelStreamInferClientMockRecorder ¶
type MockGRPCInferenceService_ModelStreamInferClientMockRecorder struct {
// contains filtered or unexported fields
}
MockGRPCInferenceService_ModelStreamInferClientMockRecorder is the mock recorder for MockGRPCInferenceService_ModelStreamInferClient.
func (*MockGRPCInferenceService_ModelStreamInferClientMockRecorder) CloseSend ¶
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) CloseSend() *gomock.Call
CloseSend indicates an expected call of CloseSend.
func (*MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Context ¶
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Context() *gomock.Call
Context indicates an expected call of Context.
func (*MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Header ¶
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Header() *gomock.Call
Header indicates an expected call of Header.
func (*MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Recv ¶
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Recv() *gomock.Call
Recv indicates an expected call of Recv.
func (*MockGRPCInferenceService_ModelStreamInferClientMockRecorder) RecvMsg ¶
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) RecvMsg(m any) *gomock.Call
RecvMsg indicates an expected call of RecvMsg.
func (*MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Send ¶
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Send(arg0 any) *gomock.Call
Send indicates an expected call of Send.
func (*MockGRPCInferenceService_ModelStreamInferClientMockRecorder) SendMsg ¶
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) SendMsg(m any) *gomock.Call
SendMsg indicates an expected call of SendMsg.
func (*MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Trailer ¶
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Trailer() *gomock.Call
Trailer indicates an expected call of Trailer.
type MockGRPCInferenceService_ModelStreamInferServer ¶
type MockGRPCInferenceService_ModelStreamInferServer struct {
// contains filtered or unexported fields
}
MockGRPCInferenceService_ModelStreamInferServer is a mock of GRPCInferenceService_ModelStreamInferServer interface.
func NewMockGRPCInferenceService_ModelStreamInferServer ¶
func NewMockGRPCInferenceService_ModelStreamInferServer(ctrl *gomock.Controller) *MockGRPCInferenceService_ModelStreamInferServer
NewMockGRPCInferenceService_ModelStreamInferServer creates a new mock instance.
func (*MockGRPCInferenceService_ModelStreamInferServer) Context ¶
func (m *MockGRPCInferenceService_ModelStreamInferServer) Context() context.Context
Context mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferServer) EXPECT ¶
func (m *MockGRPCInferenceService_ModelStreamInferServer) EXPECT() *MockGRPCInferenceService_ModelStreamInferServerMockRecorder
EXPECT returns an object that allows the caller to indicate expected use.
func (*MockGRPCInferenceService_ModelStreamInferServer) Recv ¶
func (m *MockGRPCInferenceService_ModelStreamInferServer) Recv() (*inference.ModelInferRequest, error)
Recv mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferServer) RecvMsg ¶
func (m_2 *MockGRPCInferenceService_ModelStreamInferServer) RecvMsg(m any) error
RecvMsg mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferServer) Send ¶
func (m *MockGRPCInferenceService_ModelStreamInferServer) Send(arg0 *inference.ModelStreamInferResponse) error
Send mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferServer) SendHeader ¶
func (m *MockGRPCInferenceService_ModelStreamInferServer) SendHeader(arg0 metadata.MD) error
SendHeader mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferServer) SendMsg ¶
func (m_2 *MockGRPCInferenceService_ModelStreamInferServer) SendMsg(m any) error
SendMsg mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferServer) SetHeader ¶
func (m *MockGRPCInferenceService_ModelStreamInferServer) SetHeader(arg0 metadata.MD) error
SetHeader mocks base method.
func (*MockGRPCInferenceService_ModelStreamInferServer) SetTrailer ¶
func (m *MockGRPCInferenceService_ModelStreamInferServer) SetTrailer(arg0 metadata.MD)
SetTrailer mocks base method.
type MockGRPCInferenceService_ModelStreamInferServerMockRecorder ¶
type MockGRPCInferenceService_ModelStreamInferServerMockRecorder struct {
// contains filtered or unexported fields
}
MockGRPCInferenceService_ModelStreamInferServerMockRecorder is the mock recorder for MockGRPCInferenceService_ModelStreamInferServer.
func (*MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Context ¶
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Context() *gomock.Call
Context indicates an expected call of Context.
func (*MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Recv ¶
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Recv() *gomock.Call
Recv indicates an expected call of Recv.
func (*MockGRPCInferenceService_ModelStreamInferServerMockRecorder) RecvMsg ¶
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) RecvMsg(m any) *gomock.Call
RecvMsg indicates an expected call of RecvMsg.
func (*MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Send ¶
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Send(arg0 any) *gomock.Call
Send indicates an expected call of Send.
func (*MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SendHeader ¶
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SendHeader(arg0 any) *gomock.Call
SendHeader indicates an expected call of SendHeader.
func (*MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SendMsg ¶
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SendMsg(m any) *gomock.Call
SendMsg indicates an expected call of SendMsg.
func (*MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SetHeader ¶
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SetHeader(arg0 any) *gomock.Call
SetHeader indicates an expected call of SetHeader.
func (*MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SetTrailer ¶
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SetTrailer(arg0 any) *gomock.Call
SetTrailer indicates an expected call of SetTrailer.
type MockUnsafeGRPCInferenceServiceServer ¶
type MockUnsafeGRPCInferenceServiceServer struct {
// contains filtered or unexported fields
}
MockUnsafeGRPCInferenceServiceServer is a mock of UnsafeGRPCInferenceServiceServer interface.
func NewMockUnsafeGRPCInferenceServiceServer ¶
func NewMockUnsafeGRPCInferenceServiceServer(ctrl *gomock.Controller) *MockUnsafeGRPCInferenceServiceServer
NewMockUnsafeGRPCInferenceServiceServer creates a new mock instance.
func (*MockUnsafeGRPCInferenceServiceServer) EXPECT ¶
func (m *MockUnsafeGRPCInferenceServiceServer) EXPECT() *MockUnsafeGRPCInferenceServiceServerMockRecorder
EXPECT returns an object that allows the caller to indicate expected use.
type MockUnsafeGRPCInferenceServiceServerMockRecorder ¶
type MockUnsafeGRPCInferenceServiceServerMockRecorder struct {
// contains filtered or unexported fields
}
MockUnsafeGRPCInferenceServiceServerMockRecorder is the mock recorder for MockUnsafeGRPCInferenceServiceServer.