Don't use overload for deepcopy

Author: ricardoV94

pytensor/link/numba/dispatch/basic.py

@@ -1,13 +1,11 @@
 import warnings
-from copy import copy
 from functools import singledispatch
 
 import numba
 import numpy as np
 from numba import types
 from numba.core.errors import NumbaWarning, TypingError
 from numba.cpython.unsafe.tuple import tuple_setitem  # noqa: F401
-from numba.extending import overload
 
 from pytensor import In, config
 from pytensor.compile import NUMBA
@@ -296,21 +294,21 @@ def numba_funcify_FunctionGraph(
     )
 
 
-def deepcopyop(x):
-    return copy(x)
-
+@numba_funcify.register(DeepCopyOp)
+def numba_funcify_DeepCopyOp(op, node, **kwargs):
+    if isinstance(node.inputs[0].type, TensorType):
 
-@overload(deepcopyop)
-def dispatch_deepcopyop(x):
-    if isinstance(x, types.Array):
-        return lambda x: np.copy(x)
+        @numba_njit
+        def deepcopy(x):
+            return np.copy(x)
 
-    return lambda x: x
+    else:
 
+        @numba_njit
+        def deepcopy(x):
+            return x
 
-@numba_funcify.register(DeepCopyOp)
-def numba_funcify_DeepCopyOp(op, node, **kwargs):
-    return deepcopyop
+    return deepcopy
 
 
 @numba.extending.intrinsic