From 4564776733d31b9c8bc930ab57802248aaf40b2b Mon Sep 17 00:00:00 2001 From: Ryan Lai Date: Mon, 20 Jul 2020 19:32:34 -0700 Subject: [PATCH] Close input stream too --- winml/test/common/protobufHelpers.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/winml/test/common/protobufHelpers.cpp b/winml/test/common/protobufHelpers.cpp index 420727b2ac..50299c166c 100644 --- a/winml/test/common/protobufHelpers.cpp +++ b/winml/test/common/protobufHelpers.cpp @@ -65,7 +65,7 @@ bool ProtobufHelpers::LoadOnnxTensorFromProtobufFile(onnx::TensorProto& tensor, google::protobuf::io::FileInputStream input(fd, block_size); const bool result = tensor.ParseFromZeroCopyStream(&input) && input.GetErrno() == 0; status = onnxruntime::Env::Default().FileClose(fd); - if (!result || !status.IsOK()) { + if (!result || !status.IsOK() || !input.Close()) { return false; } return true;