Skip to content

Commit

Permalink
fix filename gen in MSIMatrix.save
Browse files Browse the repository at this point in the history
  • Loading branch information
jey committed Apr 2, 2015
1 parent 40fdbbb commit 8b35dbd
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions cx_spark/src/spark_msi.py
Original file line number Diff line number Diff line change
Expand Up @@ -187,14 +187,14 @@ def cache(self):
def save(self, csvpath, metapath, name):
self.nonzeros. \
map(lambda entry: ",".join(map(str, entry))). \
saveAsTextFile(os.path.join(csvpath, name + ".csv"))
saveAsTextFile(os.path.join(csvpath, name + ".mat.csv"))
metadata = {
'dataset_shape' : self.dataset_shape,
'raw_shape' : self.raw_shape,
'shape' : self.shape,
'seen' : self.seen_bcast.value
}
with file(os.path.join(metapath, name + ".meta"), 'w') as outf:
with file(os.path.join(metapath, name + ".mat.meta"), 'w') as outf:
pickle.dump(metadata, outf)

@staticmethod
Expand Down

0 comments on commit 8b35dbd

Please sign in to comment.