Make Constant and Shared variables subclasses of the respective Varia…

…bles
ricardoV94 · Mar 24, 2024 · 60e2510 · 60e2510
1 parent 97317a5
commit 60e2510
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/pytensor/sparse/basic.py b/pytensor/sparse/basic.py
@@ -479,7 +479,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):

diff --git a/pytensor/sparse/sharedvar.py b/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

diff --git a/pytensor/tensor/sharedvar.py b/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 __getattr__(name):
@@ -31,7 +31,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.