Videre

linedance-app/venv/lib/python3.12/site-packages/numba/misc/literal.py

@@ -0,0 +1,24 @@
+from numba.core.extending import overload
+from numba.core import types
+from numba.misc.special import literally, literal_unroll
+from numba.core.errors import TypingError
+
+
+@overload(literally)
+def _ov_literally(obj):
+    if isinstance(obj, (types.Literal, types.InitialValue)):
+        return lambda obj: obj
+    else:
+        m = "Invalid use of non-Literal type in literally({})".format(obj)
+        raise TypingError(m)
+
+
+@overload(literal_unroll)
+def literal_unroll_impl(container):
+    if isinstance(container, types.Poison):
+        m = f"Invalid use of non-Literal type in literal_unroll({container})"
+        raise TypingError(m)
+
+    def impl(container):
+        return container
+    return impl