fix wired import format

SeldonIO · Oct 29, 2019 · 9dfa6ea · 9dfa6ea
1 parent a6a9f2f
commit 9dfa6ea
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/engine/src/main/java/io/seldon/engine/ExceptionControllerAdvice.java b/engine/src/main/java/io/seldon/engine/ExceptionControllerAdvice.java
@@ -18,7 +18,7 @@
 import com.google.protobuf.InvalidProtocolBufferException;
 import io.seldon.engine.exception.APIException;
 import io.seldon.engine.pb.ProtoBufUtils;
-import io.seldon.protos.PredictionProtos;
+import io.seldon.protos.PredictionProtos.SeldonMessage;
 import io.seldon.protos.PredictionProtos.Status;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
@@ -38,8 +38,7 @@ public ResponseEntity<String> handleUnauthorizedException(APIException exception
     statusBuilder.setInfo(exception.getInfo());
     statusBuilder.setStatus(Status.StatusFlag.FAILURE);
 
-    PredictionProtos.SeldonMessage message = PredictionProtos.SeldonMessage.newBuilder()
-            .setStatus(statusBuilder.build()).build();
+    SeldonMessage message = SeldonMessage.newBuilder().setStatus(statusBuilder.build()).build();
     String json;
     json = ProtoBufUtils.toJson(message);
     return new ResponseEntity<String>(