ruff fixes

Author: majosm

arraycontext/impl/pytato/compile.py

@@ -7,8 +7,6 @@
 """
 from __future__ import annotations
 
-from pytato.array import AxesT
-
 
 __copyright__ = """
 Copyright (C) 2020-1 University of Illinois Board of Trustees
@@ -63,6 +61,7 @@
     from collections.abc import Callable, Hashable, Mapping
 
     import pyopencl.array as cla
+    from pytato.array import AxesT
 
 
 logger = logging.getLogger(__name__)

arraycontext/impl/pytato/outline.py

@@ -29,25 +29,30 @@
 """
 
 import itertools
-from collections.abc import Callable, Mapping
 from dataclasses import dataclass
-from typing import cast
+from typing import TYPE_CHECKING, cast
 
 import numpy as np
 from immutabledict import immutabledict
 
 import pytato as pt
-from pymbolic import Scalar
-from pytools.tag import Tag
 
 from arraycontext.container import is_array_container_type
 from arraycontext.container.traversal import rec_keyed_map_array_container
-from arraycontext.context import (
-    Array,
-    ArrayOrContainer,
-    ArrayT,
-)
-from arraycontext.impl.pytato import _BasePytatoArrayContext
+
+
+if TYPE_CHECKING:
+    from collections.abc import Callable, Mapping
+
+    from pymbolic import Scalar
+    from pytools.tag import Tag
+
+    from arraycontext.context import (
+        Array,
+        ArrayOrContainer,
+        ArrayT,
+    )
+    from arraycontext.impl.pytato import _BasePytatoArrayContext
 
 
 def _get_arg_id_to_arg(args: tuple[object, ...],
@@ -133,7 +138,7 @@ def get_placeholder_replacement(
         if arg is None:
             return None
         elif np.isscalar(arg):
-            return cast(Scalar, arg)
+            return cast("Scalar", arg)
         elif isinstance(arg, pt.Array):
             return arg_id_to_placeholder[key]
         elif is_array_container_type(arg.__class__):

arraycontext/impl/pytato/utils.py

@@ -43,7 +43,6 @@
 from typing_extensions import override
 
 import pytools
-from pytato import AbstractResultWithNamedArrays
 from pytato.analysis import get_num_call_sites
 from pytato.array import (
     Array,
@@ -53,7 +52,6 @@
     SizeParam,
     make_placeholder,
 )
-from pytato.function import FunctionDefinition
 from pytato.target.loopy import LoopyPyOpenCLTarget
 from pytato.transform import (
     ArrayOrNames,
@@ -71,6 +69,8 @@
     from collections.abc import Mapping
 
     import loopy as lp
+    from pytato import AbstractResultWithNamedArrays
+    from pytato.function import FunctionDefinition
 
     from arraycontext import ArrayContext
     from arraycontext.impl.pytato import PytatoPyOpenCLArrayContext