From 742008e64db1b70eec22c7908132918417615b1b Mon Sep 17 00:00:00 2001
From: Shilpa Chugh <shchugh@redhat.com>
Date: Thu, 19 Sep 2024 16:21:27 +0530
Subject: [PATCH] change order

---
 tests/odh/mnist_ray_test.go | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/tests/odh/mnist_ray_test.go b/tests/odh/mnist_ray_test.go
index d2a11d9c..370dc8cf 100644
--- a/tests/odh/mnist_ray_test.go
+++ b/tests/odh/mnist_ray_test.go
@@ -259,14 +259,12 @@ func buildAndPushRayImage(test Test, namespace string, image string) {
 
 	stdoutStderr, err := cmd.CombinedOutput()
 	if err != nil {
-		fmt.Printf("print err .....")
-		fmt.Println(err)
+		fmt.Println("Error executing custom_image script :", err)
 		return
 		//log.Fatal("Error executing custom_image script :", err)
 	}
-	//test.Expect(err).NotTo(HaveOccurred())
-
 	fmt.Printf("Logs of build and custom ray image . . .\n %s", stdoutStderr)
+	test.Expect(err).NotTo(HaveOccurred())
 }
 
 func getCustomRayImage(test Test, namespace string, image string) string {