ColumnStoreReader support for intermediate keys of flat dictionaries

pull/1257/head
Greg Hogan 5 years ago
parent 1d6042c5f3
commit cf131ee8e9
  1. 7
      common/column_store.py

@ -36,7 +36,12 @@ class ColumnStoreReader():
if self._is_dict:
if not self._np_data:
self._np_data = self._load(os.path.join(self._path, 'columnstore'))
return self._np_data.keys()
# return top level keys by matching on prefix and splitting on '/'
return {
k[len(self._prefix):].split('/')[0]
for k in self._np_data.keys()
if k.startswith(self._prefix)
}
return list(self._keys)

Loading…
Cancel
Save