diff --git a/test/e2e/v1alpha3/resume-e2e-experiment.go b/test/e2e/v1alpha3/resume-e2e-experiment.go index 2b5fcf683c3..ad22c60f5d9 100644 --- a/test/e2e/v1alpha3/resume-e2e-experiment.go +++ b/test/e2e/v1alpha3/resume-e2e-experiment.go @@ -30,11 +30,13 @@ func verifyResult(exp *experimentsv1alpha3.Experiment) (*commonv1alpha3.Metric, return nil, fmt.Errorf("Best metrics not updated in status") } - metric := exp.Status.CurrentOptimalTrial.Observation.Metrics[0] - if metric.Name != exp.Spec.Objective.ObjectiveMetricName { - return nil, fmt.Errorf("Best objective metric not updated in status") + for _, metric := range exp.Status.CurrentOptimalTrial.Observation.Metrics { + if metric.Name == exp.Spec.Objective.ObjectiveMetricName { + return &metric, nil + } } - return &metric, nil + + return nil, fmt.Errorf("Best objective metric not updated in status") } func main() { diff --git a/test/e2e/v1alpha3/run-e2e-experiment.go b/test/e2e/v1alpha3/run-e2e-experiment.go index ef59292daba..f89685f0f94 100644 --- a/test/e2e/v1alpha3/run-e2e-experiment.go +++ b/test/e2e/v1alpha3/run-e2e-experiment.go @@ -30,11 +30,13 @@ func verifyResult(exp *experimentsv1alpha3.Experiment) (*commonv1alpha3.Metric, return nil, fmt.Errorf("Best metrics not updated in status") } - metric := exp.Status.CurrentOptimalTrial.Observation.Metrics[0] - if metric.Name != exp.Spec.Objective.ObjectiveMetricName { - return nil, fmt.Errorf("Best objective metric not updated in status") + for _, metric := range exp.Status.CurrentOptimalTrial.Observation.Metrics { + if metric.Name == exp.Spec.Objective.ObjectiveMetricName { + return &metric, nil + } } - return &metric, nil + + return nil, fmt.Errorf("Best objective metric not updated in status") } func main() {