From 86ec9e571ed076a62f1fa3b30056298fd68109fe Mon Sep 17 00:00:00 2001
From: Marie Generali <mgenerali@teklia.com>
Date: Wed, 30 Aug 2023 10:25:05 +0000
Subject: [PATCH] Fix parse_tokens
---
dan/utils.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/dan/utils.py b/dan/utils.py
index 5cb62ff5..12c45af4 100644
--- a/dan/utils.py
+++ b/dan/utils.py
@@ -106,8 +106,8 @@ def list_to_batches(iterable, n):
yield batch
-def parse_tokens(filename: Path) -> Dict[str, EntityType]:
+def parse_tokens(filename: str) -> Dict[str, EntityType]:
return {
name: EntityType(**tokens)
- for name, tokens in yaml.safe_load(filename.read_text()).items()
+ for name, tokens in yaml.safe_load(Path(filename).read_text()).items()
}
--
GitLab