Skip to content
Snippets Groups Projects
Commit bdd9d5bb authored by Mateusz Klimaszewski's avatar Mateusz Klimaszewski Committed by Mateusz Klimaszewski
Browse files

Fix mapping and path.

parent d711c60f
2 merge requests!9Enhanced dependency parsing develop to master,!8Enhanced dependency parsing
This commit is part of merge request !9. Comments created here will be created in the context of that merge request.
......@@ -50,6 +50,10 @@ def sentence2conllu(sentence: Sentence, keep_semrel: bool = True) -> conllu.Toke
for t in tokens:
if type(t["id"]) == list:
t["id"] = tuple(t["id"])
if t["deps"]:
for dep in t["deps"]:
if type(dep[1]) == list:
dep[1] = tuple(dep[1])
return _TokenList(tokens=tokens,
metadata=sentence.metadata)
......
......@@ -135,7 +135,7 @@ def run(_):
embeddings_dir = FLAGS.embeddings_dir
embeddings_file = None
if embeddings_dir:
embeddings_dir = embeddings_dir / language
embeddings_dir = pathlib.Path(embeddings_dir) / language
embeddings_file = [f for f in embeddings_dir.iterdir() if "vectors" in f.name and ".vec.gz" in f.name]
assert len(embeddings_file) == 1, f"Couldn't find embeddings file."
embeddings_file = embeddings_file[0]
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment