diff --git a/pkg/metricscollector/v1beta1/file-metricscollector/file-metricscollector_test.go b/pkg/metricscollector/v1beta1/file-metricscollector/file-metricscollector_test.go index 14e505f6ec4..7e8042ef66b 100644 --- a/pkg/metricscollector/v1beta1/file-metricscollector/file-metricscollector_test.go +++ b/pkg/metricscollector/v1beta1/file-metricscollector/file-metricscollector_test.go @@ -27,7 +27,6 @@ import ( commonv1beta1 "github.com/kubeflow/katib/pkg/apis/controller/common/v1beta1" v1beta1 "github.com/kubeflow/katib/pkg/apis/manager/v1beta1" "github.com/kubeflow/katib/pkg/controller.v1beta1/consts" - "google.golang.org/protobuf/testing/protocmp" ) func TestCollectObservationLog(t *testing.T) { @@ -321,7 +320,8 @@ invalid INFO {metricName: loss, metricValue: 0.3634}`, if diff := cmp.Diff(test.wantError, err, cmpopts.EquateErrors()); len(diff) != 0 { t.Errorf("Unexpected error (-want,+got):\n%s", diff) } - if diff := cmp.Diff(test.expected, actual, protocmp.Transform()); len(diff) != 0 { + observationLogCmpOpts := cmpopts.IgnoreUnexported(v1beta1.ObservationLog{}, v1beta1.MetricLog{}, v1beta1.Metric{}) + if diff := cmp.Diff(test.expected, actual, observationLogCmpOpts); len(diff) != 0 { t.Errorf("Unexpected parsed result (-want,+got):\n%s", diff) } })