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
Branches
No related tags found
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 ...@@ -50,6 +50,10 @@ def sentence2conllu(sentence: Sentence, keep_semrel: bool = True) -> conllu.Toke
for t in tokens: for t in tokens:
if type(t["id"]) == list: if type(t["id"]) == list:
t["id"] = tuple(t["id"]) 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, return _TokenList(tokens=tokens,
metadata=sentence.metadata) metadata=sentence.metadata)
......
...@@ -135,7 +135,7 @@ def run(_): ...@@ -135,7 +135,7 @@ def run(_):
embeddings_dir = FLAGS.embeddings_dir embeddings_dir = FLAGS.embeddings_dir
embeddings_file = None embeddings_file = None
if embeddings_dir: 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] 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." assert len(embeddings_file) == 1, f"Couldn't find embeddings file."
embeddings_file = embeddings_file[0] embeddings_file = embeddings_file[0]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment