Skip to content

Commit b199882

Browse files
authored
chore: pkg import only once (#1950)
Signed-off-by: guoguangwu <[email protected]>
1 parent e2d6ba4 commit b199882

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

pkg/controller.v1/common/job.go

+4-5
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,6 @@ import (
3030

3131
log "github.com/sirupsen/logrus"
3232
corev1 "k8s.io/api/core/v1"
33-
v1 "k8s.io/api/core/v1"
3433
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
3534
"k8s.io/apimachinery/pkg/runtime"
3635
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
@@ -253,7 +252,7 @@ func (jc *JobController) ReconcileJobs(
253252
queue := ""
254253
priorityClass := ""
255254
var schedulerTimeout *int32
256-
var minResources *v1.ResourceList
255+
var minResources *corev1.ResourceList
257256

258257
if runPolicy.SchedulingPolicy != nil {
259258
if minAvailable := runPolicy.SchedulingPolicy.MinAvailable; minAvailable != nil {
@@ -266,7 +265,7 @@ func (jc *JobController) ReconcileJobs(
266265
priorityClass = pc
267266
}
268267
if mr := runPolicy.SchedulingPolicy.MinResources; mr != nil {
269-
minResources = (*v1.ResourceList)(mr)
268+
minResources = (*corev1.ResourceList)(mr)
270269
}
271270
if timeout := runPolicy.SchedulingPolicy.ScheduleTimeoutSeconds; timeout != nil {
272271
schedulerTimeout = timeout
@@ -364,7 +363,7 @@ func (jc *JobController) CleanUpResources(
364363
runtimeObject runtime.Object,
365364
metaObject metav1.Object,
366365
jobStatus apiv1.JobStatus,
367-
pods []*v1.Pod,
366+
pods []*corev1.Pod,
368367
) error {
369368
if err := jc.DeletePodsAndServices(runtimeObject, runPolicy, jobStatus, pods); err != nil {
370369
return err
@@ -448,6 +447,6 @@ func (jc *JobController) CleanupJob(runPolicy *apiv1.RunPolicy, jobStatus apiv1.
448447
}
449448
}
450449

451-
func (jc *JobController) calcPGMinResources(minMember int32, replicas map[apiv1.ReplicaType]*apiv1.ReplicaSpec) *v1.ResourceList {
450+
func (jc *JobController) calcPGMinResources(minMember int32, replicas map[apiv1.ReplicaType]*apiv1.ReplicaSpec) *corev1.ResourceList {
452451
return CalcPGMinResources(minMember, replicas, jc.PriorityClassLister.Get)
453452
}

pkg/controller.v1/common/service_test.go

+2-3
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@ import (
88
apiv1 "github.com/kubeflow/training-operator/pkg/apis/kubeflow.org/v1"
99
"github.com/stretchr/testify/assert"
1010
corev1 "k8s.io/api/core/v1"
11-
v1 "k8s.io/api/core/v1"
1211
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
1312
)
1413

@@ -71,7 +70,7 @@ func TestCalculateServiceSliceSize(t *testing.T) {
7170
}
7271

7372
func TestFilterServicesForReplicaType(t *testing.T) {
74-
services := []*v1.Service{
73+
services := []*corev1.Service{
7574
{
7675
ObjectMeta: metav1.ObjectMeta{
7776
Name: "a",
@@ -110,6 +109,6 @@ func TestFilterServicesForReplicaType(t *testing.T) {
110109
if err != nil {
111110
t.Fatalf("FilterPodsForReplicaType returned error: %v", err)
112111
}
113-
want := []*v1.Service{services[0], services[2], services[4]}
112+
want := []*corev1.Service{services[0], services[2], services[4]}
114113
assert.Equal(t, want, got)
115114
}

0 commit comments

Comments
 (0)