Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions thunder/readers.py
Original file line number Diff line number Diff line change
Expand Up @@ -388,10 +388,10 @@ def getsplit(kvIter):
raise NotImplementedError("No file reader implementation for URL scheme " + scheme)

for kv in kvIter:
idx, keyName = kv
key = bucket.get_key(keyName)
idx, keyname = kv
key = bucket.get_key(keyname)
buf = key.get_contents_as_string()
yield idx, buf
yield idx, buf, keyname

npartitions = min(npartitions, self.nfiles) if npartitions else self.nfiles
rdd = self.engine.parallelize(enumerate(keylist), npartitions)
Expand All @@ -412,7 +412,7 @@ def getsplit(kv):
idx, keyName = kv
key = bucket.get_key(keyName)
buf = key.get_contents_as_string()
return idx, buf
return idx, buf, keyName

return [getsplit(kv) for kv in enumerate(keylist)]

Expand Down