fix it

kubeflow · Apr 15, 2020 · 4c7a6e3 · 4c7a6e3
1 parent 1b9b047
commit 4c7a6e3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git 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
@@ -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() {