diff --git a/dan/utils.py b/dan/utils.py index 5cb62ff5e39e6eb776eb68d4497f53a25d539144..12c45af49e881a518e637f7345567ee00888665d 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() }