diff options
Diffstat (limited to 'src/ee/tools/part_apply_souffle_post.py')
| -rw-r--r-- | src/ee/tools/part_apply_souffle_post.py | 4 |
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) |