Merge pull request #77 from mlcommons/protoio-bugfix

Add missing is_open() to ProtoInputStream
mlcommons · Dec 6, 2023 · e7d7a7c · e7d7a7c
2 parents e6e0a34 + d61b00c
commit e7d7a7c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/third_party/utils/protoio.cc b/third_party/utils/protoio.cc
@@ -155,6 +155,10 @@ void ProtoInputStream::reset() {
   createStreams();
 }
 
+bool ProtoInputStream::is_open() {
+  return fileStream.is_open();
+}
+
 bool ProtoInputStream::read(Message& msg) {
   // Read a message from the stream by getting the size, using it as
   // a limit when parsing the message, then popping the limit again

diff --git a/third_party/utils/protoio.hh b/third_party/utils/protoio.hh
@@ -143,6 +143,8 @@ class ProtoInputStream : public ProtoStream {
    */
   ~ProtoInputStream();
 
+  bool is_open();
+
   /**
    * Read a message from the stream.
    *