Add __all__ to aesara.tensor.type

aesara/tensor/__init__.py

@@ -104,162 +104,7 @@ from aesara.tensor.shape import (
 # `aesara.shared` and `tensor._shared`.
 from aesara.tensor.sort import argsort, argtopk, sort, topk, topk_and_argtopk
 from aesara.tensor.subtensor import *
-from aesara.tensor.type import (
-    TensorType,
-    bcol,
-    bmatrix,
-    brow,
-    bscalar,
-    btensor3,
-    btensor4,
-    btensor5,
-    btensor6,
-    btensor7,
-    bvector,
-    ccol,
-    cmatrix,
-    col,
-    cols,
-    complex_matrix_types,
-    complex_scalar_types,
-    complex_types,
-    complex_vector_types,
-    crow,
-    cscalar,
-    ctensor3,
-    ctensor4,
-    ctensor5,
-    ctensor6,
-    ctensor7,
-    cvector,
-    dcol,
-    dcols,
-    dmatrices,
-    dmatrix,
-    drow,
-    drows,
-    dscalar,
-    dscalars,
-    dtensor3,
-    dtensor3s,
-    dtensor4,
-    dtensor4s,
-    dtensor5,
-    dtensor5s,
-    dtensor6,
-    dtensor6s,
-    dtensor7,
-    dtensor7s,
-    dvector,
-    dvectors,
-    fcol,
-    fcols,
-    float_matrix_types,
-    float_scalar_types,
-    float_types,
-    float_vector_types,
-    fmatrices,
-    fmatrix,
-    frow,
-    frows,
-    fscalar,
-    fscalars,
-    ftensor3,
-    ftensor3s,
-    ftensor4,
-    ftensor4s,
-    ftensor5,
-    ftensor5s,
-    ftensor6,
-    ftensor6s,
-    ftensor7,
-    ftensor7s,
-    fvector,
-    fvectors,
-    icol,
-    icols,
-    imatrices,
-    imatrix,
-    int_matrix_types,
-    int_scalar_types,
-    int_types,
-    int_vector_types,
-    irow,
-    irows,
-    iscalar,
-    iscalars,
-    itensor3,
-    itensor3s,
-    itensor4,
-    itensor4s,
-    itensor5,
-    itensor5s,
-    itensor6,
-    itensor6s,
-    itensor7,
-    itensor7s,
-    ivector,
-    ivectors,
-    lcol,
-    lcols,
-    lmatrices,
-    lmatrix,
-    lrow,
-    lrows,
-    lscalar,
-    lscalars,
-    ltensor3,
-    ltensor3s,
-    ltensor4,
-    ltensor4s,
-    ltensor5,
-    ltensor5s,
-    ltensor6,
-    ltensor6s,
-    ltensor7,
-    ltensor7s,
-    lvector,
-    lvectors,
-    matrices,
-    matrix,
-    row,
-    rows,
-    scalar,
-    scalars,
-    tensor,
-    tensor3,
-    tensor3s,
-    tensor4,
-    tensor4s,
-    tensor5,
-    tensor5s,
-    tensor6,
-    tensor6s,
-    tensor7,
-    tensor7s,
-    values_eq_approx_always_true,
-    vector,
-    vectors,
-    wcol,
-    wrow,
-    wscalar,
-    wtensor3,
-    wtensor4,
-    wtensor5,
-    wtensor6,
-    wtensor7,
-    wvector,
-    zcol,
-    zmatrix,
-    zrow,
-    zscalar,
-    ztensor3,
-    ztensor4,
-    ztensor5,
-    ztensor6,
-    ztensor7,
-    zvector,
-)
+from aesara.tensor.type import *
 from aesara.tensor.type_other import *
 
 

aesara/tensor/type.py

@@ -1062,3 +1062,161 @@ def tensor7(name=None, dtype=None):
 tensor7s, ftensor7s, dtensor7s, itensor7s, ltensor7s = apply_across_args(
     tensor7, ftensor7, dtensor7, itensor7, ltensor7
 )
+
+
+__all__ = [
+    "TensorType",
+    "bcol",
+    "bmatrix",
+    "brow",
+    "bscalar",
+    "btensor3",
+    "btensor4",
+    "btensor5",
+    "btensor6",
+    "btensor7",
+    "bvector",
+    "ccol",
+    "cmatrix",
+    "col",
+    "cols",
+    "complex_matrix_types",
+    "complex_scalar_types",
+    "complex_types",
+    "complex_vector_types",
+    "crow",
+    "cscalar",
+    "ctensor3",
+    "ctensor4",
+    "ctensor5",
+    "ctensor6",
+    "ctensor7",
+    "cvector",
+    "dcol",
+    "dcols",
+    "dmatrices",
+    "dmatrix",
+    "drow",
+    "drows",
+    "dscalar",
+    "dscalars",
+    "dtensor3",
+    "dtensor3s",
+    "dtensor4",
+    "dtensor4s",
+    "dtensor5",
+    "dtensor5s",
+    "dtensor6",
+    "dtensor6s",
+    "dtensor7",
+    "dtensor7s",
+    "dvector",
+    "dvectors",
+    "fcol",
+    "fcols",
+    "float_matrix_types",
+    "float_scalar_types",
+    "float_types",
+    "float_vector_types",
+    "fmatrices",
+    "fmatrix",
+    "frow",
+    "frows",
+    "fscalar",
+    "fscalars",
+    "ftensor3",
+    "ftensor3s",
+    "ftensor4",
+    "ftensor4s",
+    "ftensor5",
+    "ftensor5s",
+    "ftensor6",
+    "ftensor6s",
+    "ftensor7",
+    "ftensor7s",
+    "fvector",
+    "fvectors",
+    "icol",
+    "icols",
+    "imatrices",
+    "imatrix",
+    "int_matrix_types",
+    "int_scalar_types",
+    "int_types",
+    "int_vector_types",
+    "irow",
+    "irows",
+    "iscalar",
+    "iscalars",
+    "itensor3",
+    "itensor3s",
+    "itensor4",
+    "itensor4s",
+    "itensor5",
+    "itensor5s",
+    "itensor6",
+    "itensor6s",
+    "itensor7",
+    "itensor7s",
+    "ivector",
+    "ivectors",
+    "lcol",
+    "lcols",
+    "lmatrices",
+    "lmatrix",
+    "lrow",
+    "lrows",
+    "lscalar",
+    "lscalars",
+    "ltensor3",
+    "ltensor3s",
+    "ltensor4",
+    "ltensor4s",
+    "ltensor5",
+    "ltensor5s",
+    "ltensor6",
+    "ltensor6s",
+    "ltensor7",
+    "ltensor7s",
+    "lvector",
+    "lvectors",
+    "matrices",
+    "matrix",
+    "row",
+    "rows",
+    "scalar",
+    "scalars",
+    "tensor",
+    "tensor3",
+    "tensor3s",
+    "tensor4",
+    "tensor4s",
+    "tensor5",
+    "tensor5s",
+    "tensor6",
+    "tensor6s",
+    "tensor7",
+    "tensor7s",
+    "values_eq_approx_always_true",
+    "vector",
+    "vectors",
+    "wcol",
+    "wrow",
+    "wscalar",
+    "wtensor3",
+    "wtensor4",
+    "wtensor5",
+    "wtensor6",
+    "wtensor7",
+    "wvector",
+    "zcol",
+    "zmatrix",
+    "zrow",
+    "zscalar",
+    "ztensor3",
+    "ztensor4",
+    "ztensor5",
+    "ztensor6",
+    "ztensor7",
+    "zvector",
+]