add resources to earlystopping container

Author: zhixian82

pkg/controller.v1beta1/suggestion/composer/composer.go

@@ -248,6 +248,7 @@ func (g *General) desiredContainers(s *suggestionsv1beta1.Suggestion,
 					ContainerPort: consts.DefaultEarlyStoppingPort,
 				},
 			},
+			Resources: earlyStoppingConfigData.Resource,
 		}
 
 		containers = append(containers, earlyStoppingContainer)

pkg/util/v1beta1/katibconfig/config.go

@@ -46,8 +46,9 @@ type SuggestionConfig struct {
 
 // EarlyStoppingConfig is the JSON early stopping structure in Katib config.
 type EarlyStoppingConfig struct {
-	Image           string            `json:"image"`
-	ImagePullPolicy corev1.PullPolicy `json:"imagePullPolicy,omitempty"`
+	Image           string                      `json:"image"`
+	ImagePullPolicy corev1.PullPolicy           `json:"imagePullPolicy,omitempty"`
+	Resource        corev1.ResourceRequirements `json:"resources,omitempty"`
 }
 
 // MetricsCollectorConfig is the JSON metrics collector structure in Katib config.
@@ -177,6 +178,9 @@ func GetEarlyStoppingConfigData(algorithmName string, client client.Client) (Ear
 	// Set Image Pull Policy.
 	earlyStoppingConfigData.ImagePullPolicy = setImagePullPolicy(earlyStoppingConfigData.ImagePullPolicy)
 
+	// Set resource requirements
+	earlyStoppingConfigData.Resource = setResourceRequirements(earlyStoppingConfigData.Resource)
+
 	return earlyStoppingConfigData, nil
 }
 

pkg/util/v1beta1/katibconfig/config_test.go

@@ -426,6 +426,7 @@ func newFakeEarlyStoppingConfig() *EarlyStoppingConfig {
 	return &EarlyStoppingConfig{
 		Image:           "early-stopping-image",
 		ImagePullPolicy: consts.DefaultImagePullPolicy,
+		Resource:        *setFakeResourceRequirements(),
 	}
 }