-gpuConfig, _ = utils.GetGPUConfigBySKU(ragEngineObj.Spec.Compute.InstanceType)
+gpuConfig, err = utils.GetGPUConfigBySKU(ragEngineObj.Spec.Compute.InstanceType)
+if err != nil {
+    // Log the error or handle it appropriately
+    skuNumGPUs = 0
+}

-time.Sleep(30 * time.Second)
+// Example of polling with a timeout
+ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+defer cancel()
 
+for ctx.Err() == nil {
+    // Check condition
+    if conditionMet {
+        break
+    }
+    time.Sleep(1 * time.Second) // Adjust sleep duration as needed
+}
+
+if ctx.Err() != nil {
+    t.Errorf("Timed out waiting for condition")
+}
+

-return "", fmt.Errorf("no node containing 'ragpool' found")
+// Example of retrying the node lookup
+attempts := 5
+for i := 0; i < attempts; i++ {
+    nodeList := &v1.NodeList{}
+    err := utils.TestingCluster.KubeClient.List(ctx, nodeList)
+    if err != nil {
+        return "", fmt.Errorf("failed to list nodes: %v", err)
+    }
 
+    for _, node := range nodeList.Items {
+        if strings.Contains(node.Name, "ragpool") {
+            return node.Name, nil
+        }
+    }
+    time.Sleep(5 * time.Second) // Adjust sleep duration as needed
+}
+return "", fmt.Errorf("no node containing 'ragpool' found after %d attempts", attempts)
+

-skuNumGPUs = 0
+skuNumGPUs = 1 // or another appropriate default value