forked from kubeflow/training-operator
-
Notifications
You must be signed in to change notification settings - Fork 0
/
xgboost_types.go
87 lines (75 loc) · 3.27 KB
/
xgboost_types.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
// Copyright 2021 The Kubeflow Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package v1
import (
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)
// EDIT THIS FILE! THIS IS SCAFFOLDING FOR YOU TO OWN!
// NOTE: json tags are required. Any new fields you add must have json tags for the fields to be serialized.
const (
// XGBoostJobDefaultPortName is name of the port used to communicate between Master and Workers.
XGBoostJobDefaultPortName = "xgboostjob-port"
// XGBoostJobDefaultContainerName is the name of the XGBoostJob container.
XGBoostJobDefaultContainerName = "xgboost"
// XGBoostJobDefaultPort is default value of the port.
XGBoostJobDefaultPort = 9999
// XGBoostJobDefaultRestartPolicy is default RestartPolicy for XGBReplicaSpecs.
XGBoostJobDefaultRestartPolicy = RestartPolicyNever
// XGBoostJobKind is the kind name.
XGBoostJobKind = "XGBoostJob"
// XGBoostJobPlural is the XGBoostJobPlural for XGBoostJob.
XGBoostJobPlural = "xgboostjobs"
// XGBoostJobSingular is the singular for XGBoostJob.
XGBoostJobSingular = "xgboostjob"
// XGBoostJobFrameworkName is the name of the ML Framework
XGBoostJobFrameworkName = "xgboost"
// XGBoostJobReplicaTypeMaster is the type for master replica.
XGBoostJobReplicaTypeMaster ReplicaType = "Master"
// XGBoostJobReplicaTypeWorker is the type for worker replicas.
XGBoostJobReplicaTypeWorker ReplicaType = "Worker"
)
// XGBoostJobSpec defines the desired state of XGBoostJob
type XGBoostJobSpec struct {
// INSERT ADDITIONAL SPEC FIELDS - desired state of cluster
// Important: Run "make" to regenerate code after modifying this file
//+kubebuilder:validation:Optional
RunPolicy RunPolicy `json:"runPolicy"`
XGBReplicaSpecs map[ReplicaType]*ReplicaSpec `json:"xgbReplicaSpecs"`
}
//+kubebuilder:object:root=true
//+kubebuilder:subresource:status
//+kubebuilder:printcolumn:name="State",type=string,JSONPath=`.status.conditions[-1:].type`
//+kubebuilder:printcolumn:name="Age",type=date,JSONPath=`.metadata.creationTimestamp`
// +genclient
// +k8s:deepcopy-gen:interfaces=k8s.io/apimachinery/pkg/runtime.Object
// XGBoostJob is the Schema for the xgboostjobs API
// +k8s:openapi-gen=true
type XGBoostJob struct {
metav1.TypeMeta `json:",inline"`
metav1.ObjectMeta `json:"metadata,omitempty"`
Spec XGBoostJobSpec `json:"spec,omitempty"`
Status JobStatus `json:"status,omitempty"`
}
//+kubebuilder:object:root=true
// +k8s:deepcopy-gen:interfaces=k8s.io/apimachinery/pkg/runtime.Object
// XGBoostJobList contains a list of XGBoostJob
type XGBoostJobList struct {
metav1.TypeMeta `json:",inline"`
metav1.ListMeta `json:"metadata,omitempty"`
Items []XGBoostJob `json:"items"`
}
func init() {
SchemeBuilder.Register(&XGBoostJob{}, &XGBoostJobList{})
SchemeBuilder.SchemeBuilder.Register(addXGBoostJobDefaultingFuncs)
}