Fixed: Inv Numba implementation now correctly returns non-integral inverses of…

aesara/link/numba/dispatch/scalar.py

@@ -21,6 +21,7 @@ from aesara.scalar.basic import (
     Clip,
     Composite,
     Identity,
+    Inv,
     Mul,
     ScalarOp,
     Second,
@@ -170,3 +171,12 @@ def numba_funcify_Second(op, node, **kwargs):
         return y
 
     return second
+
+
+@numba_funcify.register(Inv)
+def numba_funcify_Inv(op, node, **kwargs):
+    @numba.njit(inline="always")
+    def inv(x):
+        return 1 / x
+
+    return inv

tests/link/test_numba.py

@@ -796,6 +796,25 @@ def test_Cast(v, dtype):
     )
 
 
+@pytest.mark.parametrize(
+    "v, dtype",
+    [
+        (set_test_value(aet.iscalar(), np.array(10, dtype="int32")), aesb.float64),
+    ],
+)
+def test_Inv(v, dtype):
+    g = aesb.inv(v)
+    g_fg = FunctionGraph(outputs=[g])
+    compare_numba_and_py(
+        g_fg,
+        [
+            i.tag.test_value
+            for i in g_fg.inputs
+            if not isinstance(i, (SharedVariable, Constant))
+        ],
+    )
+
+
 @pytest.mark.parametrize(
     "v, shape, ndim",
     [