From a8d5a666250934229631a3592cfd1bf499acdc67 Mon Sep 17 00:00:00 2001
From: manonBlanco <blanco@teklia.com>
Date: Tue, 21 May 2024 10:51:05 +0200
Subject: [PATCH] Create missing folders if needed
---
dan/datasets/entities/extract.py | 2 ++
dan/datasets/tokens/generate.py | 1 +
2 files changed, 3 insertions(+)
diff --git a/dan/datasets/entities/extract.py b/dan/datasets/entities/extract.py
index c6d23d36..6991ddba 100644
--- a/dan/datasets/entities/extract.py
+++ b/dan/datasets/entities/extract.py
@@ -18,6 +18,8 @@ def run(database: Path, output_file: Path) -> None:
entities = list(
map(itemgetter(0), EntityType.select(EntityType.name).distinct().tuples())
)
+
+ output_file.parent.mkdir(exist_ok=True, parents=True)
output_file.write_text(
yaml.safe_dump({"entities": entities}, explicit_start=True, allow_unicode=True)
)
diff --git a/dan/datasets/tokens/generate.py b/dan/datasets/tokens/generate.py
index 6906e262..c815ade0 100644
--- a/dan/datasets/tokens/generate.py
+++ b/dan/datasets/tokens/generate.py
@@ -37,6 +37,7 @@ def run(entities: Path, end_tokens: bool, output_file: Path) -> None:
}
# Save entities & tokens to YAML
+ output_file.parent.mkdir(exist_ok=True, parents=True)
output_file.write_text(
yaml.safe_dump(tokens, explicit_start=True, allow_unicode=True, sort_keys=False)
)
--
GitLab