Make Constant and Shared variables subclasses of the basic Variable types

Author: ricardoV94

pytensor/sparse/basic.py

@@ -478,7 +478,7 @@ def pytensor_hash(self):
         return hash_from_sparse(d)
 
 
-class SparseConstant(TensorConstant, _sparse_py_operators):
+class SparseConstant(SparseVariable, TensorConstant):
     format = property(lambda self: self.type.format)
 
     def signature(self):

pytensor/sparse/sharedvar.py

@@ -3,11 +3,11 @@
 import scipy.sparse
 
 from pytensor.compile import shared_constructor
-from pytensor.sparse.basic import SparseTensorType, _sparse_py_operators
+from pytensor.sparse.basic import SparseTensorType, SparseVariable
 from pytensor.tensor.sharedvar import TensorSharedVariable
 
 
-class SparseTensorSharedVariable(TensorSharedVariable, _sparse_py_operators):
+class SparseTensorSharedVariable(TensorSharedVariable, SparseVariable):
     @property
     def format(self):
         return self.type.format

pytensor/tensor/sharedvar.py

@@ -6,7 +6,7 @@
 from pytensor.misc.safe_asarray import _asarray
 from pytensor.tensor import _get_vector_length
 from pytensor.tensor.type import TensorType
-from pytensor.tensor.variable import _tensor_py_operators
+from pytensor.tensor.variable import TensorVariable
 
 
 def load_shared_variable(val):
@@ -19,7 +19,7 @@ def load_shared_variable(val):
     return tensor_constructor(val)
 
 
-class TensorSharedVariable(_tensor_py_operators, SharedVariable):
+class TensorSharedVariable(SharedVariable, TensorVariable):
     def zero(self, borrow: bool = False):
         r"""Set the values of a shared variable to 0.