bug fix: wrong open mode

usc-isi-i2 · Sep 6, 2018 · 389fae7 · 389fae7
1 parent 38a1547
commit 389fae7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/rltk/io/io_utils.py b/rltk/io/io_utils.py
@@ -1,17 +1,18 @@
 import io
 
 
-def get_file_handler(f):
+def get_file_handler(f, mode='r'):
     """
     Helper function for getting file handler.
 
     Args:
         f (Union[str,io.IOBase]): File path or handler.
+        mode (str, optional): Parameter mode in :py:meth:`open`. Defaults to `r`.
 
     Returns:
         io.IOBase: File handler.
     """
     if isinstance(f, io.IOBase):
         return f
 
-    return open(f, 'w')
+    return open(f, mode)
diff --git a/rltk/io/writer/ground_truth_writer.py b/rltk/io/writer/ground_truth_writer.py
@@ -12,7 +12,7 @@ class GroundTruthWriter(Writer):
         file_handler (io.IOBase): It can be file name or file handler.
     """
     def __init__(self, file_handler):
-        self._file_handler = get_file_handler(file_handler)
+        self._file_handler = get_file_handler(file_handler, 'w')
         fieldnames = ['id1', 'id2', 'label']
         self._csv_writer = csv.DictWriter(self._file_handler, fieldnames=fieldnames)
         self._csv_writer.writeheader()