diff --git a/doc/reference.rst b/doc/reference.rst index c743d5834728cb845c0fb19b78534bf39f3d5202..a05ca62b0d70020632efb52a606b8a48b547c747 100644 --- a/doc/reference.rst +++ b/doc/reference.rst @@ -3,4 +3,4 @@ Reference .. automodule:: pytato.array .. automodule:: pytato.program -.. automodule:: pytato.symbolic +.. automodule:: pytato.scalar_expr diff --git a/pytato/array.py b/pytato/array.py index 061bf823f11fb74fe5537ca040e275c05134ffd5..f0dd76cdad37640c86c9e01fa81fde3b39c6fa46 100644 --- a/pytato/array.py +++ b/pytato/array.py @@ -87,8 +87,8 @@ Node constructors such as :class:`Placeholder.__init__` and import numpy as np import pymbolic.primitives as prim -import pytato.symbolic as sym -from pytato.symbolic import ScalarExpression +import pytato.scalar_expr as scalar_expr +from pytato.scalar_expr import ScalarExpression from dataclasses import dataclass from pytools import is_single_valued @@ -267,7 +267,7 @@ def _check_identifier(s: str, ns: Optional[Namespace] = None) -> bool: return True -class _ShapeChecker(sym.WalkMapper): +class _ShapeChecker(scalar_expr.WalkMapper): def __init__(self, ns: Optional[Namespace] = None): super().__init__() self.ns = ns @@ -288,7 +288,7 @@ def normalize_shape( def normalize_shape_component( s: ConvertibleToShapeComponent) -> ScalarExpression: if isinstance(s, str): - s = sym.parse(s) + s = scalar_expr.parse(s) if isinstance(s, int): if s < 0: @@ -301,7 +301,7 @@ def normalize_shape( return s if isinstance(shape, str): - shape = sym.parse(shape) + shape = scalar_expr.parse(shape) from numbers import Number if isinstance(shape, (Number, prim.Expression)): diff --git a/pytato/symbolic.py b/pytato/scalar_expr.py similarity index 100% rename from pytato/symbolic.py rename to pytato/scalar_expr.py