diff --git a/doc/reference.rst b/doc/reference.rst index a991d2823f06a46aed4f824fbc28fdb1d358fb8a..2a14d6a247b503185119602e4077ba732d84b27b 100644 --- a/doc/reference.rst +++ b/doc/reference.rst @@ -3,5 +3,5 @@ Reference .. automodule:: pytato.array .. automodule:: pytato.program -.. automodule:: pytato.array_expr .. automodule:: pytato.scalar_expr +.. automodule:: pytato.transform diff --git a/pytato/array.py b/pytato/array.py index 096cfc2e0b5c00c0112321c4b49bb2ae775e0577..54f6e32594e2f5893005dae63f5f46fd6eb846bb 100644 --- a/pytato/array.py +++ b/pytato/array.py @@ -179,7 +179,7 @@ class Namespace(Mapping[str, "Array"]): return len(self._symbol_table) def copy(self) -> Namespace: - from pytato.array_expr import CopyMapper, copy_namespace + from pytato.transform import CopyMapper, copy_namespace return copy_namespace(self, CopyMapper(Namespace())) def assign(self, name: str, value: Array) -> str: diff --git a/pytato/array_expr.py b/pytato/transform.py similarity index 97% rename from pytato/array_expr.py rename to pytato/transform.py index fdd82028d312234f0654ca2908e6cbbdfde96fa7..76a1b4990bc1fc43d28302d449d40b6702e6920a 100644 --- a/pytato/array_expr.py +++ b/pytato/transform.py @@ -31,10 +31,10 @@ from pymbolic.mapper import Mapper as MapperBase from pytato.array import Array, IndexLambda, Namespace, Output, Placeholder __doc__ = """ -.. currentmodule:: pytato.array_expr +.. currentmodule:: pytato.transform -Tools for Array Expressions ---------------------------- +Transforming Computations +------------------------- .. autoclass:: CopyMapper .. autofunction:: copy_namespace