fix recordio

python/mxnet/recordio.py

@@ -59,6 +59,7 @@ def close(self):
             check_call(_LIB.MXRecordIOWriterFree(self.handle))
         else:
             check_call(_LIB.MXRecordIOReaderFree(self.handle))
+        self.is_open = False
 
     def reset(self):
         """Reset pointer to first item. If record is opened with 'w',
@@ -135,6 +136,8 @@ def open(self):
                 self.keys.append(key)
 
     def close(self):
+        if not self.is_open:
+            return
         super(MXIndexedRecordIO, self).close()
         self.fidx.close()
 

src/c_api/c_api.cc

@@ -723,6 +723,7 @@ int MXRecordIOWriterFree(RecordIOHandle handle) {
     reinterpret_cast<MXRecordIOContext*>(handle);
   delete context->writer;
   delete context->stream;
+  delete context;
   API_END();
 }
 
@@ -763,6 +764,7 @@ int MXRecordIOReaderFree(RecordIOHandle handle) {
   delete context->reader;
   delete context->stream;
   delete context->read_buff;
+  delete context;
   API_END();
 }