Minor fixes

inducer · Nov 14, 2023 · 4dc68ba · 4dc68ba
1 parent 6c6d042
commit 4dc68ba
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/meshmode/discretization/connection/direct.py b/meshmode/discretization/connection/direct.py
@@ -700,7 +700,7 @@ def group_pick_knl(is_surjective: bool):
                                 from_element_indices[iel],
                                 dof_pick_lists[dof_pick_list_indices[iel], idof]
                             ]
-                        { if_present })
+                        {if_present})
                 """,
                 [
                     lp.GlobalArg("ary", None,

diff --git a/meshmode/discretization/poly_element.py b/meshmode/discretization/poly_element.py
@@ -28,7 +28,6 @@
 from warnings import warn
 
 import numpy as np
-from numpy.typing import ArrayLike
 from pytools import memoize_method, memoize_on_first_arg
 from meshmode.mesh import (
         MeshElementGroup as _MeshElementGroup,
@@ -522,7 +521,7 @@ class TensorProductElementGroupBase(PolynomialElementGroupBase,
         HypercubeElementGroupBase):
     def __init__(self, mesh_el_group: _MeshTensorProductElementGroup,
                  order: int, index=None, *, basis: Basis,
-                 unit_nodes: ArrayLike):
+                 unit_nodes: np.ndarray):
         """
         :arg basis: a :class:`modepy.TensorProductBasis`.
         :arg unit_nodes: unit nodes for the tensor product, obtained by
@@ -537,10 +536,11 @@ def __init__(self, mesh_el_group: _MeshTensorProductElementGroup,
         if isinstance(basis, mp.TensorProductBasis):
             for b in basis.bases:
                 if b._dim != 1:
-                    raise NotImplementedError("All bases used to construct the tensor "
-                                     "product must be of dimension 1. Support "
-                                     "for higher-dimensional component bases "
-                                     "does not yet exist.")
+                    raise NotImplementedError(
+                        "All bases used to construct the tensor "
+                        "product must be of dimension 1. Support "
+                        "for higher-dimensional component bases "
+                        "does not yet exist.")
 
         if unit_nodes.shape[0] != mesh_el_group.dim:
             raise ValueError("unit node dimension does not match element group: "