diff --git a/sumpy/tools.py b/sumpy/tools.py index be404fe9d0dfff22117db8c76064f7ddee1fe941..32b4d3eecb0268263e2f631d17b7beacbc6d90a9 100644 --- a/sumpy/tools.py +++ b/sumpy/tools.py @@ -24,31 +24,25 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. """ -from pytools import memoize_method -from pytools.tag import Tag, tag_dataclass -import numbers -import warnings -import os -import sys import enum +import logging +import warnings from abc import ABC, abstractmethod from dataclasses import dataclass +from typing import TYPE_CHECKING, Any, List, Optional, Sequence, Tuple +import loopy as lp +import numpy as np +from arraycontext import Array from pymbolic.mapper import WalkMapper -import pymbolic +from pytools import T, memoize_method +from pytools.tag import Tag, tag_dataclass -import numpy as np import sumpy.symbolic as sym -import pyopencl as cl -import pyopencl.array as cla - -import loopy as lp -from typing import Any, List, Optional, TYPE_CHECKING - -import logging -logger = logging.getLogger(__name__) if TYPE_CHECKING: + import pyopencl as cl + from sumpy.kernel import Kernel