aboutsummaryrefslogtreecommitdiff
path: root/src/ee/tools
diff options
context:
space:
mode:
Diffstat (limited to 'src/ee/tools')
-rw-r--r--src/ee/tools/part_apply_souffle_post.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/ee/tools/part_apply_souffle_post.py b/src/ee/tools/part_apply_souffle_post.py
index 8630603..8ef7c6a 100644
--- a/src/ee/tools/part_apply_souffle_post.py
+++ b/src/ee/tools/part_apply_souffle_post.py
@@ -15,10 +15,12 @@ def work(in_path: Path, out_path: Path, work_dir: Path):
in_parts: ObjDb[Part] = ObjDb[Part]()
uri_idx = in_parts.add_unique_index("uri", lambda p: p.uri)
- for entry in load_db(in_path).parts:
+ in_db = load_db(in_path)
+ for entry in in_db.parts:
in_parts.add(Part(entry.part))
out_parts = PartDb()
+ out_parts.categories.extend(in_db.categories)
for part in in_parts:
out_parts.add_entry(part, False)