diff --git a/nd4j/nd4j-serde/nd4j-kryo/src/main/java/org/nd4j/Nd4jSerializer.java b/nd4j/nd4j-serde/nd4j-kryo/src/main/java/org/nd4j/Nd4jSerializer.java index 568f5ab98..e3e2a447b 100644 --- a/nd4j/nd4j-serde/nd4j-kryo/src/main/java/org/nd4j/Nd4jSerializer.java +++ b/nd4j/nd4j-serde/nd4j-kryo/src/main/java/org/nd4j/Nd4jSerializer.java @@ -70,11 +70,7 @@ public class Nd4jSerializer extends Serializer { @Override public INDArray read(Kryo kryo, Input input, Class type) { DataInputStream dis = new DataInputStream(input); - try { - return Nd4j.read(dis); - } catch (IOException e) { - throw new RuntimeException(e); - } + return Nd4j.read(dis); //Note: input should NOT be closed manually here - may be needed elsewhere (and closing here will cause serialization to fail) }