Skip to content

Commit f6d1f32

Browse files
author
Frankie Robertson
committed
Fix pickle dumping in train_sif
1 parent f2f5ede commit f6d1f32

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train_sif.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def train_sif(corpus, out_path):
4242
root, paths = get_eval_paths(corpus)
4343
for vec in ["numberbatch", "fasttext", "word2vec", "triple"]:
4444
pc = train_one_sif(open(paths["train"]["suptag"], "rb"), vec)
45-
pickle.dump(pc, open(sif_filename(out_path, vec, "w")))
45+
pickle.dump(pc, open(sif_filename(out_path, vec), "wb"))
4646

4747

4848
def load_sif(path, vec):

0 commit comments

Comments
 (0)