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
...@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment