Add NumOfHosts to WorkerGroupSpec (CRD change only)

docs/reference/api.md

@@ -216,6 +216,7 @@ _Appears in:_
 | `replicas` _integer_ | Replicas is the number of desired Pods for this worker group. See https://github.com/ray-project/kuberay/pull/1443 for more details about the reason for making this field optional. |
 | `minReplicas` _integer_ | MinReplicas denotes the minimum number of desired Pods for this worker group. |
 | `maxReplicas` _integer_ | MaxReplicas denotes the maximum number of desired Pods for this worker group, and the default value is maxInt32. |
+| `numOfHosts` _integer_ | NumOfHosts denotes the number of hosts to create per replica. The default value is 1. |
 | `rayStartParams` _object (keys:string, values:string)_ | RayStartParams are the params of the start command: address, object-store-memory, ... |
 | `template` _[PodTemplateSpec](https://kubernetes.io/docs/reference/generated/kubernetes-api/v1.28/#podtemplatespec-v1-core)_ | Template is a pod template for the worker |
 | `scaleStrategy` _[ScaleStrategy](#scalestrategy)_ | ScaleStrategy defines which pods to remove |

ray-operator/apis/ray/v1/raycluster_types.go

@@ -56,6 +56,9 @@ type WorkerGroupSpec struct {
 	// MaxReplicas denotes the maximum number of desired Pods for this worker group, and the default value is maxInt32.
 	// +kubebuilder:default:=2147483647
 	MaxReplicas *int32 `json:"maxReplicas"`
+	// NumOfHosts denotes the number of hosts to create per replica. The default value is 1.
+	// +kubebuilder:default:=1
+	NumOfHosts int32 `json:"numOfHosts,omitempty"`
 	// RayStartParams are the params of the start command: address, object-store-memory, ...
 	RayStartParams map[string]string `json:"rayStartParams"`
 	// Template is a pod template for the worker

ray-operator/apis/ray/v1/raycluster_types_test.go

@@ -58,6 +58,7 @@ var myRayCluster = &RayCluster{
 				Replicas:    pointer.Int32(3),
 				MinReplicas: pointer.Int32(0),
 				MaxReplicas: pointer.Int32(10000),
+				NumOfHosts:  1,
 				GroupName:   "small-group",
 				RayStartParams: map[string]string{
 					"port":                        "6379",

ray-operator/apis/ray/v1/rayjob_types_test.go

@@ -90,6 +90,7 @@ var expectedRayJob = RayJob{
 					Replicas:    pointer.Int32(3),
 					MinReplicas: pointer.Int32(0),
 					MaxReplicas: pointer.Int32(10000),
+					NumOfHosts:  1,
 					GroupName:   "small-group",
 					RayStartParams: map[string]string{
 						"port":     "6379",
@@ -213,7 +214,8 @@ var testRayJobJSON = `{
                     "replicas": 3,
                     "minReplicas": 0,
                     "maxReplicas": 10000,
-                    "rayStartParams": {
+		    "numOfHosts": 1,
+		    "rayStartParams": {
                         "num-cpus": "1",
                         "port": "6379"
                     },

ray-operator/apis/ray/v1/rayservice_types_test.go

@@ -90,6 +90,7 @@ var myRayService = &RayService{
 					Replicas:    pointer.Int32(3),
 					MinReplicas: pointer.Int32(0),
 					MaxReplicas: pointer.Int32(10000),
+					NumOfHosts:  1,
 					GroupName:   "small-group",
 					RayStartParams: map[string]string{
 						"port":                        "6379",
@@ -211,7 +212,8 @@ var expected = `{
                "replicas":3,
                "minReplicas":0,
                "maxReplicas":10000,
-               "rayStartParams":{
+	       "numOfHosts":1,
+	       "rayStartParams":{
                   "dashboard-agent-listen-port":"52365",
                   "num-cpus":"1",
                   "port":"6379"