testing

package
v1.9.1 Latest Latest
Warning

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

Go to latest
Published: Mar 24, 2025 License: Apache-2.0 Imports: 21 Imported by: 0

Documentation

Index

Constants

View Source
const (
	NotFoundError errorMatcher = iota
	ForbiddenError
	InvalidError
)

Variables

This section is empty.

Functions

func AsIndex

func AsIndex(builder *fake.ClientBuilder) client.FieldIndexer

func BeAPIError

func BeAPIError(name errorMatcher) types.GomegaMatcher

func BeForbiddenError

func BeForbiddenError() types.GomegaMatcher

func BeInvalidError

func BeInvalidError() types.GomegaMatcher

func BeNotFoundError

func BeNotFoundError() types.GomegaMatcher

func NewClientBuilder

func NewClientBuilder(addToSchemes ...func(s *runtime.Scheme) error) *fake.ClientBuilder

Types

type ClusterTrainingRuntimeWrapper

type ClusterTrainingRuntimeWrapper struct {
	kubeflowv2.ClusterTrainingRuntime
}

func MakeClusterTrainingRuntimeWrapper

func MakeClusterTrainingRuntimeWrapper(name string) *ClusterTrainingRuntimeWrapper

func (*ClusterTrainingRuntimeWrapper) Obj

func (*ClusterTrainingRuntimeWrapper) RuntimeSpec

type JobSetWrapper

type JobSetWrapper struct {
	jobsetv1alpha2.JobSet
}

func MakeJobSetWrapper

func MakeJobSetWrapper(namespace, name string) *JobSetWrapper

func (*JobSetWrapper) Annotation

func (j *JobSetWrapper) Annotation(key, value string) *JobSetWrapper

func (*JobSetWrapper) Conditions

func (j *JobSetWrapper) Conditions(conditions ...metav1.Condition) *JobSetWrapper

func (*JobSetWrapper) ContainerTrainer

func (j *JobSetWrapper) ContainerTrainer(image string, command []string, args []string, res corev1.ResourceList) *JobSetWrapper

func (*JobSetWrapper) ContainerTrainerEnv

func (j *JobSetWrapper) ContainerTrainerEnv(env []corev1.EnvVar) *JobSetWrapper

func (*JobSetWrapper) ContainerTrainerPorts

func (j *JobSetWrapper) ContainerTrainerPorts(ports []corev1.ContainerPort) *JobSetWrapper

func (*JobSetWrapper) ControllerReference

func (j *JobSetWrapper) ControllerReference(gvk schema.GroupVersionKind, name, uid string) *JobSetWrapper

func (*JobSetWrapper) InitContainerDatasetInitializerEnv

func (j *JobSetWrapper) InitContainerDatasetInitializerEnv(env []corev1.EnvVar) *JobSetWrapper

func (*JobSetWrapper) InitContainerDatasetInitializerEnvFrom

func (j *JobSetWrapper) InitContainerDatasetInitializerEnvFrom(envFrom []corev1.EnvFromSource) *JobSetWrapper

func (*JobSetWrapper) InitContainerDatasetModelInitializer

func (j *JobSetWrapper) InitContainerDatasetModelInitializer(image string, command []string, args []string, res corev1.ResourceList) *JobSetWrapper

func (*JobSetWrapper) InitContainerModelInitializerEnv

func (j *JobSetWrapper) InitContainerModelInitializerEnv(env []corev1.EnvVar) *JobSetWrapper

func (*JobSetWrapper) InitContainerModelInitializerEnvFrom

func (j *JobSetWrapper) InitContainerModelInitializerEnvFrom(envFrom []corev1.EnvFromSource) *JobSetWrapper

func (*JobSetWrapper) Label

func (j *JobSetWrapper) Label(key, value string) *JobSetWrapper

func (*JobSetWrapper) NumNodes

func (j *JobSetWrapper) NumNodes(numNodes int32) *JobSetWrapper

func (*JobSetWrapper) Obj

func (*JobSetWrapper) PodLabel

func (j *JobSetWrapper) PodLabel(key, value string) *JobSetWrapper

func (*JobSetWrapper) Replicas

func (j *JobSetWrapper) Replicas(replicas int32) *JobSetWrapper

func (*JobSetWrapper) Suspend

func (j *JobSetWrapper) Suspend(suspend bool) *JobSetWrapper

type SchedulerPluginsPodGroupWrapper

type SchedulerPluginsPodGroupWrapper struct {
	schedulerpluginsv1alpha1.PodGroup
}

func MakeSchedulerPluginsPodGroup

func MakeSchedulerPluginsPodGroup(namespace, name string) *SchedulerPluginsPodGroupWrapper

func (*SchedulerPluginsPodGroupWrapper) ControllerReference

func (*SchedulerPluginsPodGroupWrapper) MinMember

func (*SchedulerPluginsPodGroupWrapper) MinResources

func (*SchedulerPluginsPodGroupWrapper) Obj

func (*SchedulerPluginsPodGroupWrapper) SchedulingTimeout

type TrainJobDatasetConfigWrapper

type TrainJobDatasetConfigWrapper struct {
	kubeflowv2.DatasetConfig
}

func MakeTrainJobDatasetConfigWrapper

func MakeTrainJobDatasetConfigWrapper() *TrainJobDatasetConfigWrapper

func (*TrainJobDatasetConfigWrapper) ContainerEnv

func (*TrainJobDatasetConfigWrapper) Obj

func (*TrainJobDatasetConfigWrapper) SecretRef

func (*TrainJobDatasetConfigWrapper) StorageUri

type TrainJobModelConfigWrapper

type TrainJobModelConfigWrapper struct {
	kubeflowv2.ModelConfig
}

func MakeTrainJobModelConfigWrapper

func MakeTrainJobModelConfigWrapper() *TrainJobModelConfigWrapper

func (*TrainJobModelConfigWrapper) ContainerEnv

func (*TrainJobModelConfigWrapper) Obj

func (*TrainJobModelConfigWrapper) SecretRef

func (*TrainJobModelConfigWrapper) StorageUri

type TrainJobTrainerWrapper

type TrainJobTrainerWrapper struct {
	kubeflowv2.Trainer
}

func MakeTrainJobTrainerWrapper

func MakeTrainJobTrainerWrapper() *TrainJobTrainerWrapper

func (*TrainJobTrainerWrapper) Container

func (t *TrainJobTrainerWrapper) Container(image string, command []string, args []string, resRequests corev1.ResourceList) *TrainJobTrainerWrapper

func (*TrainJobTrainerWrapper) ContainerEnv

func (*TrainJobTrainerWrapper) NumNodes

func (t *TrainJobTrainerWrapper) NumNodes(numNodes int32) *TrainJobTrainerWrapper

func (*TrainJobTrainerWrapper) NumProcPerNode

func (t *TrainJobTrainerWrapper) NumProcPerNode(numProcPerNode string) *TrainJobTrainerWrapper

func (*TrainJobTrainerWrapper) Obj

type TrainJobWrapper

type TrainJobWrapper struct {
	kubeflowv2.TrainJob
}

func MakeTrainJobWrapper

func MakeTrainJobWrapper(namespace, name string) *TrainJobWrapper

func (*TrainJobWrapper) DatasetConfig

func (t *TrainJobWrapper) DatasetConfig(datasetConfig *kubeflowv2.DatasetConfig) *TrainJobWrapper

func (*TrainJobWrapper) ManagedBy

func (t *TrainJobWrapper) ManagedBy(m string) *TrainJobWrapper

func (*TrainJobWrapper) ModelConfig

func (t *TrainJobWrapper) ModelConfig(modelConfig *kubeflowv2.ModelConfig) *TrainJobWrapper

func (*TrainJobWrapper) Obj

func (*TrainJobWrapper) RuntimeRef

func (t *TrainJobWrapper) RuntimeRef(gvk schema.GroupVersionKind, name string) *TrainJobWrapper

func (*TrainJobWrapper) SpecAnnotation

func (t *TrainJobWrapper) SpecAnnotation(key, value string) *TrainJobWrapper

func (*TrainJobWrapper) SpecLabel

func (t *TrainJobWrapper) SpecLabel(key, value string) *TrainJobWrapper

func (*TrainJobWrapper) Suspend

func (t *TrainJobWrapper) Suspend(suspend bool) *TrainJobWrapper

func (*TrainJobWrapper) Trainer

func (t *TrainJobWrapper) Trainer(trainer *kubeflowv2.Trainer) *TrainJobWrapper

func (*TrainJobWrapper) UID

func (t *TrainJobWrapper) UID(uid string) *TrainJobWrapper

type TrainingRuntimeSpecWrapper

type TrainingRuntimeSpecWrapper struct {
	kubeflowv2.TrainingRuntimeSpec
}

func (*TrainingRuntimeSpecWrapper) ContainerTrainer

func (s *TrainingRuntimeSpecWrapper) ContainerTrainer(image string, command []string, args []string, res corev1.ResourceList) *TrainingRuntimeSpecWrapper

func (*TrainingRuntimeSpecWrapper) ContainerTrainerEnv

func (s *TrainingRuntimeSpecWrapper) ContainerTrainerEnv(env []corev1.EnvVar) *TrainingRuntimeSpecWrapper

func (*TrainingRuntimeSpecWrapper) InitContainerDatasetModelInitializer

func (s *TrainingRuntimeSpecWrapper) InitContainerDatasetModelInitializer(image string, command []string, args []string, res corev1.ResourceList) *TrainingRuntimeSpecWrapper

func (*TrainingRuntimeSpecWrapper) NumNodes

func (*TrainingRuntimeSpecWrapper) Obj

func (*TrainingRuntimeSpecWrapper) PodGroupPolicyCoscheduling

func (*TrainingRuntimeSpecWrapper) PodGroupPolicyCoschedulingSchedulingTimeout

func (s *TrainingRuntimeSpecWrapper) PodGroupPolicyCoschedulingSchedulingTimeout(timeout int32) *TrainingRuntimeSpecWrapper

func (*TrainingRuntimeSpecWrapper) TorchPolicy

func (s *TrainingRuntimeSpecWrapper) TorchPolicy(numNodes int32, numProcPerNode string) *TrainingRuntimeSpecWrapper

type TrainingRuntimeWrapper

type TrainingRuntimeWrapper struct {
	kubeflowv2.TrainingRuntime
}

func MakeTrainingRuntimeWrapper

func MakeTrainingRuntimeWrapper(namespace, name string) *TrainingRuntimeWrapper

func (*TrainingRuntimeWrapper) Annotation

func (r *TrainingRuntimeWrapper) Annotation(key, value string) *TrainingRuntimeWrapper

func (*TrainingRuntimeWrapper) Label

func (*TrainingRuntimeWrapper) Obj

func (*TrainingRuntimeWrapper) RuntimeSpec

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL