reorder imports

Signed-off-by: Kevin <[email protected]>

Author: KPostOffice

tests/common/resources/custom-nb-small.yaml

@@ -55,7 +55,7 @@ spec:
         - name: PIP_TRUSTED_HOST
           value: {{.PipTrustedHost}}
         image: {{.NotebookImage}}
-        command: ["/bin/sh", "-c", "pip install papermill && papermill /opt/app-root/notebooks/{{.NotebookConfigMapFileName}} /opt/app-root/src/mcad-out.ipynb -p namespace {{.Namespace}} -p openshift_api_url {{.OpenShiftApiUrl}} -p kubernetes_user_bearer_token {{.KubernetesUserBearerToken}} -p num_gpus {{ .NumGpus }} --log-output && sleep infinity"]
+        command: {{.Command}}
         # args: ["pip install papermill && oc login --token=${OCP_TOKEN} --server=${OCP_SERVER} --insecure-skip-tls-verify=true && papermill /opt/app-root/notebooks/mcad.ipynb /opt/app-root/src/mcad-out.ipynb" ]
         imagePullPolicy: Always
         # livenessProbe:

tests/kfto/kfto_mnist_sdk_test.go

@@ -21,10 +21,11 @@ import (
 	"testing"
 
 	. "github.com/onsi/gomega"
-	. "github.com/opendatahub-io/distributed-workloads/tests/common"
 	. "github.com/project-codeflare/codeflare-common/support"
 
 	v1 "k8s.io/api/core/v1"
+
+	. "github.com/opendatahub-io/distributed-workloads/tests/common"
 )
 
 func TestMnistSDK(t *testing.T) {

tests/odh/mnist_ray_test.go

@@ -23,14 +23,15 @@ import (
 	"time"
 
 	. "github.com/onsi/gomega"
-	. "github.com/opendatahub-io/distributed-workloads/tests/common"
 	. "github.com/project-codeflare/codeflare-common/support"
 	rayv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1"
 
 	corev1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"sigs.k8s.io/kueue/apis/kueue/v1beta1"
+
+	. "github.com/opendatahub-io/distributed-workloads/tests/common"
 )
 
 func TestMnistRayCpu(t *testing.T) {

tests/odh/mnist_raytune_hpo_test.go

@@ -23,14 +23,15 @@ import (
 	"time"
 
 	. "github.com/onsi/gomega"
-	. "github.com/opendatahub-io/distributed-workloads/tests/common"
 	. "github.com/project-codeflare/codeflare-common/support"
 	rayv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1"
 
 	corev1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"sigs.k8s.io/kueue/apis/kueue/v1beta1"
+
+	. "github.com/opendatahub-io/distributed-workloads/tests/common"
 )
 
 func TestMnistRayTuneHpoCpu(t *testing.T) {

tests/odh/ray_finetune_llm_deepspeed_test.go

@@ -24,11 +24,12 @@ import (
 	"time"
 
 	. "github.com/onsi/gomega"
-	. "github.com/opendatahub-io/distributed-workloads/tests/common"
 	. "github.com/project-codeflare/codeflare-common/support"
 	rayv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1"
 
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	. "github.com/opendatahub-io/distributed-workloads/tests/common"
 )
 
 func TestRayFinetuneLlmDeepspeedDemoLlama_2_7b(t *testing.T) {

tests/odh/raytune_oai_mr_grpc_test.go

@@ -26,11 +26,12 @@ import (
 	"time"
 
 	. "github.com/onsi/gomega"
-	. "github.com/opendatahub-io/distributed-workloads/tests/common"
 	. "github.com/project-codeflare/codeflare-common/support"
 	rayv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1"
 
 	corev1 "k8s.io/api/core/v1"
+
+	. "github.com/opendatahub-io/distributed-workloads/tests/common"
 )
 
 func TestRaytuneOaiMrGrpcCpu(t *testing.T) {