Rename core Conv1d to Convolve1d

Author: ricardoV94

pytensor/link/jax/dispatch/signal/conv.py

@@ -1,11 +1,11 @@
 import jax
 
 from pytensor.link.jax.dispatch import jax_funcify
-from pytensor.tensor.signal.conv import Conv1d
+from pytensor.tensor.signal.conv import Convolve1d
 
 
-@jax_funcify.register(Conv1d)
-def jax_funcify_Conv1d(op, node, **kwargs):
+@jax_funcify.register(Convolve1d)
+def jax_funcify_Convolve1d(op, node, **kwargs):
     mode = op.mode
 
     def conv1d(data, kernel):

pytensor/link/numba/dispatch/signal/conv.py

@@ -2,11 +2,11 @@
 
 from pytensor.link.numba.dispatch import numba_funcify
 from pytensor.link.numba.dispatch.basic import numba_njit
-from pytensor.tensor.signal.conv import Conv1d
+from pytensor.tensor.signal.conv import Convolve1d
 
 
-@numba_funcify.register(Conv1d)
-def numba_funcify_Conv1d(op, node, **kwargs):
+@numba_funcify.register(Convolve1d)
+def numba_funcify_Convolve1d(op, node, **kwargs):
     mode = op.mode
 
     @numba_njit

pytensor/tensor/signal/conv.py

@@ -15,7 +15,7 @@
     from pytensor.tensor import TensorLike
 
 
-class Conv1d(Op):
+class Convolve1d(Op):
     __props__ = ("mode",)
     gufunc_signature = "(n),(k)->(o)"
 
@@ -129,4 +129,4 @@ def convolve1d(
         )
         mode = "valid"
 
-    return cast(TensorVariable, Blockwise(Conv1d(mode=mode))(in1, in2))
+    return cast(TensorVariable, Blockwise(Convolve1d(mode=mode))(in1, in2))

tests/tensor/signal/test_conv.py

@@ -8,7 +8,7 @@
 from pytensor.graph import ancestors, rewrite_graph
 from pytensor.tensor import matrix, vector
 from pytensor.tensor.blockwise import Blockwise
-from pytensor.tensor.signal.conv import Conv1d, convolve1d
+from pytensor.tensor.signal.conv import Convolve1d, convolve1d
 from tests import unittest_tools as utt
 
 
@@ -81,4 +81,4 @@ def test_convolve1d_batch_graph(mode):
         if var.owner is not None and isinstance(var.owner.op, Blockwise)
     ]
     # Check any Blockwise are just Conv1d
-    assert all(isinstance(node.op.core_op, Conv1d) for node in blockwise_nodes)
+    assert all(isinstance(node.op.core_op, Convolve1d) for node in blockwise_nodes)