Fix closing recordio files from pyapi (#436)

google · Dec 20, 2019 · 551f3f4 · 551f3f4
1 parent 3000f6f
commit 551f3f4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sling/pyapi/pyrecordio.cc b/sling/pyapi/pyrecordio.cc
@@ -67,7 +67,7 @@ int PyRecordReader::Init(PyObject *args, PyObject *kwds) {
   if (!CheckIO(File::Open(filename, "r", &f))) return -1;
 
   // Create record reader.
-  reader = new RecordReader(filename, options);
+  reader = new RecordReader(f, options);
   return 0;
 }
 
@@ -276,7 +276,7 @@ int PyRecordWriter::Init(PyObject *args, PyObject *kwds) {
   if (!CheckIO(File::Open(filename, "w", &f))) return -1;
 
   // Create record writer.
-  writer = new RecordWriter(filename, options);
+  writer = new RecordWriter(f, options);
 
   return 0;
 }