fix after rebase

SeldonIO · May 18, 2020 · dc6e7ea · dc6e7ea
1 parent 2f14964
commit dc6e7ea
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/operator/apis/machinelearning.seldon.io/v1/seldondeployment_webhook_test.go b/operator/apis/machinelearning.seldon.io/v1/seldondeployment_webhook_test.go
@@ -399,7 +399,7 @@ func TestOverrideMetricsPortName(t *testing.T) {
 		Predictors: []PredictorSpec{
 			{
 				Name: "p1",
-				Graph: &PredictiveUnit{
+				Graph: PredictiveUnit{
 					Name:           "classifier",
 					Implementation: &impl,
 				},
@@ -418,7 +418,7 @@ func TestOverrideMetricsPortName(t *testing.T) {
 	g.Expect(defaultMetricsPort).To(BeNil())
 
 	// Graph
-	pu := GetPredictiveUnit(spec.Predictors[0].Graph, "classifier")
+	pu := GetPredictiveUnit(&spec.Predictors[0].Graph, "classifier")
 	g.Expect(pu).ToNot(BeNil())
 	g.Expect(pu.Endpoint.Type).To(Equal(REST))
 	g.Expect(*pu.Type).To(Equal(MODEL))
@@ -933,4 +933,4 @@ func TestPredictorNoGraph(t *testing.T) {
 	spec.DefaultSeldonDeployment("mydep", "default")
 	err := spec.ValidateSeldonDeployment()
 	g.Expect(err).ToNot(BeNil())
-}
+}