Merge pull request #7546 from stuartarchibald/fix/7545

Fix handling of missing const key in LiteralStrKeyDict

numba/core/typing/builtins.py

@@ -645,7 +645,10 @@ def generic(self, args, kws):
         if not isinstance(tup, types.LiteralStrKeyDict):
             return
         if isinstance(idx, str):
-            lookup = tup.fields.index(idx)
+            if idx in tup.fields:
+                lookup = tup.fields.index(idx)
+            else:
+                raise errors.NumbaKeyError(f"Key '{idx}' is not in dict.")
             ret = tup.types[lookup]
         if ret is not None:
             sig = signature(ret, *args)

numba/tests/test_dictobject.py

@@ -2251,6 +2251,18 @@ def foo():
 
         foo()
 
+    def test_const_key_not_in_dict(self):
+
+        @njit
+        def foo():
+            a = {'not_a': 2j, 'c': 'd', 'e': np.zeros(4)}
+            return a['a']
+
+        with self.assertRaises(TypingError) as raises:
+            foo()
+
+        self.assertIn("Key 'a' is not in dict.", str(raises.exception))
+
     def test_uncommon_identifiers(self):
         # Tests uncommon identifiers like numerical values and operators in
         # the key fields. See #6518 and #7416.