gci format

flyteorg · Jul 26, 2024 · d2d9ba4 · d2d9ba4
1 parent 7c821dc
commit d2d9ba4
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 13 deletions.
diff --git a/flyteplugins/go/tasks/plugins/k8s/ray/batchscheduler/default.go b/flyteplugins/go/tasks/plugins/k8s/ray/batchscheduler/default.go
@@ -1,9 +1,10 @@
 package batchscheduler
 
 import (
-	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/plugins"
 	v1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/plugins"
 )
 
 var (
@@ -27,6 +28,6 @@ func (d *DefaultPlugin) ParseJob(
 ) error {
 	return nil
 }
-func (d *DefaultPlugin) ProcessHead(metadata *metav1.ObjectMeta, head *v1.PodSpec) {}
+func (d *DefaultPlugin) ProcessHead(metadata *metav1.ObjectMeta, head *v1.PodSpec)                {}
 func (d *DefaultPlugin) ProcessWorker(metadata *metav1.ObjectMeta, worker *v1.PodSpec, index int) {}
-func (d *DefaultPlugin) AfterProcess(metadata *metav1.ObjectMeta) {}
+func (d *DefaultPlugin) AfterProcess(metadata *metav1.ObjectMeta)                                 {}
diff --git a/flyteplugins/go/tasks/plugins/k8s/ray/batchscheduler/plugins.go b/flyteplugins/go/tasks/plugins/k8s/ray/batchscheduler/plugins.go
@@ -1,9 +1,10 @@
 package batchscheduler
 
 import (
-	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/plugins"
 	v1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/plugins"
 )
 
 type SchedulerPlugin interface {

diff --git a/flyteplugins/go/tasks/plugins/k8s/ray/batchscheduler/plugins_test.go b/flyteplugins/go/tasks/plugins/k8s/ray/batchscheduler/plugins_test.go
@@ -7,18 +7,18 @@ import (
 )
 
 func TestCreateSchedulerPlugin(t *testing.T) {
-	var tests = []struct{
-		input *Config
+	var tests = []struct {
+		input  *Config
 		expect string
 	}{
 		{input: &Config{Scheduler: DefaultScheduler}, expect: DefaultScheduler},
 		{input: &Config{Scheduler: Yunikorn}, expect: Yunikorn},
-		{input: &Config{Scheduler:"Unknown"}, expect: DefaultScheduler},
+		{input: &Config{Scheduler: "Unknown"}, expect: DefaultScheduler},
 	}
 	for _, tt := range tests {
 		t.Run("New scheduler plugin", func(t *testing.T) {
 			p := NewSchedulerPlugin(tt.input)
 			assert.Equal(t, tt.expect, p.GetSchedulerName())
 		})
 	}
-}
+}
diff --git a/flyteplugins/go/tasks/plugins/k8s/ray/batchscheduler/yunikorn.go b/flyteplugins/go/tasks/plugins/k8s/ray/batchscheduler/yunikorn.go
@@ -4,9 +4,10 @@ import (
 	"encoding/json"
 	"fmt"
 
-	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/plugins"
 	v1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/plugins"
 )
 
 const (
@@ -20,7 +21,7 @@ const (
 
 type YunikornGangSchedulingConfig struct {
 	Annotations map[string]map[string]string
-	Parameters string
+	Parameters  string
 }
 
 func NewYunikornPlugin() *YunikornGangSchedulingConfig {

diff --git a/flyteplugins/go/tasks/plugins/k8s/ray/batchscheduler/yunikorn_test.go b/flyteplugins/go/tasks/plugins/k8s/ray/batchscheduler/yunikorn_test.go
@@ -1,15 +1,15 @@
 package batchscheduler
 
 import (
-	"testing"
 	"encoding/json"
+	"testing"
 
 	"github.com/stretchr/testify/assert"
-
-	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/plugins"
 	v1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/plugins"
 )
 
 var (