More useful message when caling a Map

connorjward · Apr 25, 2024 · a910cb2 · a910cb2
1 parent 44e1ff2
commit a910cb2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/pyop3/ir/lower.py b/pyop3/ir/lower.py
@@ -450,8 +450,8 @@ def compile(expr: Instruction, name="mykernel"):
             textwrap.dedent(
                 """
                 #include <stdlib.h>
-                
-                
+
+
                 int32_t cmpfunc(const void * a, const void * b) {
                    return ( *(int32_t*)a - *(int32_t*)b );
                 }

diff --git a/pyop3/itree/tree.py b/pyop3/itree/tree.py
@@ -567,11 +567,14 @@ def __init__(self, connectivity, name=None, *, numbering=None) -> None:
 
     def __call__(self, index):
         if isinstance(index, (ContextFreeIndex, ContextFreeCalledMap)):
-            leaf_target_paths = tuple(
-                freeze({mcpt.target_axis: mcpt.target_component})
-                for path in index.leaf_target_paths
-                for mcpt in self.connectivity[path]
-            )
+            try:
+                leaf_target_paths = tuple(
+                    freeze({mcpt.target_axis: mcpt.target_component})
+                    for path in index.leaf_target_paths
+                    for mcpt in self.connectivity[path]
+                )
+            except KeyError:
+                raise KeyError(f"Map does not have a suitable 'from index' that matches the provided index: {index.leaf_target_paths}")
             return ContextFreeCalledMap(self, index, leaf_target_paths)
         else:
             return CalledMap(self, index)