Skip to content

Commit

Permalink
override orginal size in prepare for msturing-10M-clustered
Browse files Browse the repository at this point in the history
  • Loading branch information
harsha-simhadri committed Jul 12, 2023
1 parent d29ff1a commit 9d16e59
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions benchmark/datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -168,8 +168,7 @@ def prepare(self, skip_data=False, original_size=10**9):
download(sourceurl, outfile, max_size=file_size)
# then overwrite the header...
header = np.memmap(outfile, shape=2, dtype='uint32', mode="r+")
print (header)
print(original_size)

assert header[0] == original_size
assert header[1] == self.d
header[0] = self.nb
Expand Down

0 comments on commit 9d16e59

Please sign in to comment.