fix merge impl

NREL · Apr 18, 2023 · 402b416 · 402b416
1 parent 77c6ed4
commit 402b416
Show file tree

Hide file tree

Showing 2 changed files with 20,070 additions and 1,191 deletions.
diff --git a/examples/nrel_pnnl/merge/merge.py b/examples/nrel_pnnl/merge/merge.py
@@ -140,11 +140,14 @@ def get_initial_mapping(g1, g2, typ, ns):
         inp = input("> ")
         if inp == 'done':
             return mapping
+        if not inp:
+            continue
         parts = inp.split(' ')
         g1idx, g2idx = int(parts[0]), int(parts[1])
         mapping[g1ents[g1idx].uri] = g2ents[g2idx].uri
         g1ents.pop(g1idx)
         g2ents.pop(g2idx)
+    return mapping
 
 
 def get_pairs_by_type(g1, g2, typ, ns, mapping=None) -> List[Tuple[EntityFeature, EntityFeature]]: