v1

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

	Spec   ModelSpec   `json:"spec,omitempty"`
	Status ModelStatus `json:"status,omitempty"`
}

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

type ModelSpec struct {
	Containers []*commonv1.ContainerDefinition `json:"containers,omitempty"`

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

	ExecutionRoleArn *string `json:"executionRoleArn"`

	PrimaryContainer *commonv1.ContainerDefinition `json:"primaryContainer,omitempty"`

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

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

	Region *string `json:"region"`

	SageMakerEndpoint *string `json:"sageMakerEndpoint,omitempty"`
}

type ModelStatus struct {

	// The status of the model.
	Status string `json:"status,omitempty"`

	// The name of the model in SageMaker.
	SageMakerModelName string `json:"sageMakerModelName,omitempty"`

	// The last time this status was updated.
	LastCheckTime *metav1.Time `json:"lastUpdateTime,omitempty"`

	// The Model ARN of the SageMaker model
	ModelArn string `json:"modelArn,omitempty"`

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