fix file iter to convert int id to str

usc-isi-i2 · Apr 19, 2017 · 3bd1769 · 3bd1769
1 parent 43ac6ff
commit 3bd1769
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/rltk/core.py b/rltk/core.py
@@ -575,7 +575,9 @@ def get_file_iterator(self, file_path, *args, **kwargs):
                 For `csv` file, `id_column` and `value_columns` (list) should be set. \
                 if there's no header, please set `field_names` (list).
         Returns:
-            misc, list: id, value list.
+            str, list: id, value list. \
+                If the extracted id is a int, it will be convert to string with a 'int-' prefix,\
+                for json line only supports string as key in object.
         """
         return FileIterator(file_path=self._get_abs_path(file_path), *args, **kwargs)
 

diff --git a/rltk/file_iterator.py b/rltk/file_iterator.py
@@ -63,6 +63,9 @@ def next(self):
                 value = [line]
 
             self._count += 1
+            # id should be a string
+            if isinstance(oid, int):
+                oid = 'int-' + str(oid)
             return oid, value
 
         except StopIteration as e: