Make Clip return a Numba scalar

aesara/link/numba/dispatch.py

@@ -1183,9 +1183,15 @@ def numba_funcify_Clip(op, **kwargs):
     @numba.njit
     def clip(_x, _min, _max):
         x = to_scalar(_x)
-        min = to_scalar(_min)
-        max = to_scalar(_max)
-        return np.where(x < min, min, to_scalar(np.where(x > max, max, x)))
+        _min_scalar = to_scalar(_min)
+        _max_scalar = to_scalar(_max)
+
+        if x < _min_scalar:
+            return _min_scalar
+        elif x > _max_scalar:
+            return _max_scalar
+        else:
+            return x
 
     return clip
 

tests/link/test_numba.py

@@ -917,6 +917,17 @@ def test_Clip(v, min, max):
     )
 
 
+def test_scalar_Elemwise_Clip():
+    a = aet.scalar("a")
+    b = aet.scalar("b")
+
+    z = aet.switch(1, a, b)
+    c = aet.clip(z, 1, 3)
+    c_fg = FunctionGraph(outputs=[c])
+
+    compare_numba_and_py(c_fg, [1, 1])
+
+
 @pytest.mark.parametrize(
     "vals, dtype",
     [