fix: Add parallel trial count

Signed-off-by: Ce Gao <gaoce@caicloud.io>
kubeflow · Sep 24, 2019 · 7f27eb2 · 7f27eb2
1 parent 52d0985
commit 7f27eb2
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/controller.v1alpha3/suggestion/suggestionclient/suggestionclient.go b/pkg/controller.v1alpha3/suggestion/suggestionclient/suggestionclient.go
@@ -147,6 +147,12 @@ func (g *General) ConvertExperiment(e *experimentsv1alpha3.Experiment) *suggesti
 			Parameters: convertParameters(e.Spec.Parameters),
 		},
 	}
+	if e.Spec.ParallelTrialCount != nil {
+		res.Spec.ParallelTrialCount = *e.Spec.ParallelTrialCount
+	}
+	if e.Spec.MaxTrialCount != nil {
+		res.Spec.MaxTrialCount = *e.Spec.MaxTrialCount
+	}
 	return res
 }