Create HasDataType mixin

aesara/graph/type.py

@@ -267,6 +267,12 @@ class Type(MetaObject):
         return cls.values_eq(a, b)
 
 
+class HasDataType:
+    """A mixing for an `Op` type that has a :attr:`dtype` attribute."""
+
+    dtype: str
+
+
 class CType(Type, CLinkerType):
     """Convenience wrapper combining `Type` and `CLinkerType`.
 

aesara/sparse/type.py

@@ -2,7 +2,7 @@ import numpy as np
 import scipy.sparse
 
 import aesara
-from aesara.graph.type import Type
+from aesara.graph.type import HasDataType, Type
 
 
 def _is_sparse(x):
@@ -24,7 +24,7 @@ def _is_sparse(x):
     return isinstance(x, scipy.sparse.spmatrix)
 
 
-class SparseType(Type):
+class SparseType(Type, HasDataType):
     """
     Fundamental way to create a sparse node.
 

aesara/tensor/type.py

@@ -8,7 +8,7 @@ import aesara
 from aesara import scalar as aes
 from aesara.configdefaults import config
 from aesara.graph.basic import Variable
-from aesara.graph.type import CType
+from aesara.graph.type import CType, HasDataType
 from aesara.misc.safe_asarray import _asarray
 from aesara.utils import apply_across_args
 
@@ -46,7 +46,7 @@ dtype_specs_map = {
 }
 
 
-class TensorType(CType):
+class TensorType(CType, HasDataType):
     r"""Symbolic `Type` representing `numpy.ndarray`\s."""
 
     __props__ = ("dtype", "shape")