Adjust SDK version, fix TestMNISTRayClusterSDK

Author: sutaakar

Makefile

@@ -30,7 +30,7 @@ KUBERAY_VERSION ?= v0.5.0
 RAY_VERSION ?= 2.5.0
 
 # CODEFLARE_SDK_VERSION defines the default version of the CodeFlare SDK
-CODEFLARE_SDK_VERSION ?= 0.4.4
+CODEFLARE_SDK_VERSION ?= 0.6.1
 
 # OPERATORS_REPO_ORG points to GitHub repository organization where bundle PR is opened against
 # OPERATORS_REPO_FORK_ORG points to GitHub repository fork organization where bundle build is pushed to

test/e2e/mnist_raycluster_sdk.py

@@ -12,13 +12,12 @@
 cluster = Cluster(ClusterConfiguration(
     name='mnist',
     namespace=namespace,
-    min_worker=1,
-    max_worker=1,
+    num_workers=1,
     min_cpus='500m',
     max_cpus=1,
     min_memory=0.5,
     max_memory=1,
-    gpu=0,
+    num_gpus=0,
     instascale=False,
 ))
 

test/e2e/mnist_raycluster_sdk_test.go

@@ -45,6 +45,9 @@ func TestMNISTRayClusterSDK(t *testing.T) {
 		test.T().Skip("Requires https://github.com/project-codeflare/codeflare-sdk/pull/146")
 	}
 
+	// Currently blocked by https://github.com/project-codeflare/codeflare-sdk/pull/271 , remove the skip once SDK with the PR is released
+	test.T().Skip("Requires https://github.com/project-codeflare/codeflare-sdk/pull/271")
+
 	// Create a namespace
 	namespace := test.NewTestNamespace()
 

test/support/defaults.go

@@ -5,7 +5,7 @@ package support
 // ***********************
 
 const (
-	CodeFlareSDKVersion = "0.4.4"
+	CodeFlareSDKVersion = "0.6.1"
 	RayVersion          = "2.5.0"
 	RayImage            = "rayproject/ray:2.5.0"
 )