v1

type TrainingJob struct {
	metav1.TypeMeta   `json:",inline"`
	metav1.ObjectMeta `json:"metadata,omitempty"`

	Spec   TrainingJobSpec   `json:"spec"`
	Status TrainingJobStatus `json:"status,omitempty"`
}

type TrainingJobList struct {
	metav1.TypeMeta `json:",inline"`
	metav1.ListMeta `json:"metadata,omitempty"`
	Items           []TrainingJob `json:"items"`
}

type TrainingJobSpec struct {
	AlgorithmSpecification *commonv1.AlgorithmSpecification `json:"algorithmSpecification"`

	EnableInterContainerTrafficEncryption *bool `json:"enableInterContainerTrafficEncryption,omitempty"`

	EnableNetworkIsolation *bool `json:"enableNetworkIsolation,omitempty"`

	EnableManagedSpotTraining *bool `json:"enableManagedSpotTraining,omitempty"`

	HyperParameters []*commonv1.KeyValuePair `json:"hyperParameters,omitempty"`

	// +kubebuilder:validation:MinItems=1
	InputDataConfig []commonv1.Channel `json:"inputDataConfig,omitempty"`

	OutputDataConfig *commonv1.OutputDataConfig `json:"outputDataConfig"`

	CheckpointConfig *commonv1.CheckpointConfig `json:"checkpointConfig,omitempty"`

	ResourceConfig *commonv1.ResourceConfig `json:"resourceConfig"`

	// +kubebuilder:validation:MinLength=20
	RoleArn *string `json:"roleArn"`

	// +kubebuilder:validation:MinLength=1
	Region *string `json:"region"`

	// A custom SageMaker endpoint to use when communicating with SageMaker.
	// +kubebuilder:validation:Pattern=^(https|http)://.*$
	SageMakerEndpoint *string `json:"sageMakerEndpoint,omitempty"`

	StoppingCondition *commonv1.StoppingCondition `json:"stoppingCondition"`

	DebugRuleConfigurations []*commonv1.DebugRuleConfiguration `json:"debugRuleConfigurations,omitempty"`

	DebugHookConfig *commonv1.DebugHookConfig `json:"debugHookConfig,omitempty"`

	TensorBoardOutputConfig *commonv1.TensorBoardOutputConfig `json:"tensorBoardOutputConfig,omitempty"`

	Tags []commonv1.Tag `json:"tags,omitempty"`

	// The SageMaker training job name. This is optional for the SageMaker K8s operator. If it is empty,
	// the operator will populate it with a generated name.
	// +kubebuilder:validation:MaxLength=63
	TrainingJobName *string `json:"trainingJobName,omitempty"`

	VpcConfig *commonv1.VpcConfig `json:"vpcConfig,omitempty"`
}

type TrainingJobStatus struct {
	// The status of the training job.
	// https://docs.aws.amazon.com/sagemaker/latest/dg/API_DescribeTrainingJob.html#SageMaker-DescribeTrainingJob-response-TrainingJobStatus
	TrainingJobStatus string `json:"trainingJobStatus,omitempty"`

	// The secondary, more granular status of the training job.
	// https://docs.aws.amazon.com/sagemaker/latest/dg/API_DescribeTrainingJob.html#SageMaker-DescribeTrainingJob-response-SecondaryStatus
	SecondaryStatus string `json:"secondaryStatus,omitempty"`

	// Field to store additional information, for example if
	// we are unable to check the status we update this.
	Additional string `json:"additional,omitempty"`

	// The last time that we checked the status of the SageMaker job.
	LastCheckTime *metav1.Time `json:"lastCheckTime,omitempty"`

	// Cloud Watch url for training log
	CloudWatchLogUrl string `json:"cloudWatchLogUrl,omitempty"`

	//SageMaker training job name
	SageMakerTrainingJobName string `json:"sageMakerTrainingJobName,omitempty"`

	//Full path to the training artifact (model)
	ModelPath string `json:"modelPath,omitempty"`

	// Status of rule evaluation jobs, obtained from DebugRuleEvaluationStatuses.
	// https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_DescribeTrainingJob.html#sagemaker-DescribeTrainingJob-response-DebugRuleEvaluationStatuses
	DebugRuleEvaluationStatuses []commonv1.DebugRuleEvaluationStatus `json:"debugRuleEvaluationStatuses,omitempty"`
}