Skip to content

Commit ec2e65c

Browse files
committed
feat: support reconciling suggestion volume regardless of resumePolicy for pbt
1 parent 3b0fcd2 commit ec2e65c

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

examples/v1beta1/hp-tuning/simple-pbt.yaml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@ spec:
88
maxTrialCount: 2
99
parallelTrialCount: 2
1010
maxFailedTrialCount: 3
11-
resumePolicy: FromVolume
1211
objective:
1312
type: maximize
1413
goal: 0.99

pkg/controller.v1beta1/suggestion/suggestion_controller.go

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ import (
4747
const (
4848
ControllerName = "suggestion-controller"
4949
)
50+
var ReconcileSuggestionVolumeOverrides = [...] string {"pbt"}
5051

5152
var log = logf.Log.WithName(ControllerName)
5253

@@ -191,8 +192,17 @@ func (r *ReconcileSuggestion) ReconcileSuggestion(instance *suggestionsv1beta1.S
191192
suggestionNsName := types.NamespacedName{Name: instance.GetName(), Namespace: instance.GetNamespace()}
192193
logger := log.WithValues("Suggestion", suggestionNsName)
193194

195+
// Check if requested algorithm is in reconcile override list
196+
isReconcileOverride := false
197+
for _, algorithmName := range ReconcileSuggestionVolumeOverrides {
198+
if algorithmName == instance.Spec.Algorithm.AlgorithmName {
199+
isReconcileOverride = true
200+
break
201+
}
202+
}
203+
194204
// If ResumePolicy = FromVolume volume is reconciled for suggestion
195-
if instance.Spec.ResumePolicy == experimentsv1beta1.FromVolume {
205+
if isReconcileOverride || instance.Spec.ResumePolicy == experimentsv1beta1.FromVolume {
196206
pvc, pv, err := r.DesiredVolume(instance)
197207
if err != nil {
198208
return err

0 commit comments

Comments
 (0)