From f77b3dc6773de1f6fd08c8791b0311978d7da1d7 Mon Sep 17 00:00:00 2001 From: kendricktan Date: Thu, 20 Oct 2016 23:40:51 +1000 Subject: [PATCH] Fixed train_parser examples when model_dir isn't None --- examples/training/train_parser.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/training/train_parser.py b/examples/training/train_parser.py index 7e4ed90d0..daf23cc49 100644 --- a/examples/training/train_parser.py +++ b/examples/training/train_parser.py @@ -27,10 +27,10 @@ def train_parser(nlp, train_data, left_labels, right_labels): def main(model_dir=None): if model_dir is not None: - model_dir = pathlb.Path(model_dir) + model_dir = pathlib.Path(model_dir) if not model_dir.exists(): model_dir.mkdir() - assert model_dir.isdir() + assert model_dir.is_dir() nlp = spacy.load('en', tagger=False, parser=False, entity=False, vectors=False) @@ -62,7 +62,7 @@ def main(model_dir=None): print(word.text, word.dep_, word.head.text) if model_dir is not None: - with (model_dir / 'config.json').open('wb') as file_: + with (model_dir / 'config.json').open('w') as file_: json.dump(parser.cfg, file_) parser.model.dump(str(model_dir / 'model'))