Merge pull request #228 from scientificcomputing/dokken/v0.9.0-compat-entitymap-typehint

Make EntityMap typehint backwards compatible (v0.9.x)

Author: jorgensd

src/scifem/interpolation.py

@@ -4,6 +4,7 @@
 import numpy as np
 import numpy.typing as npt
 from .utils import unroll_dofmap
+from .mesh import _EntityMap
 
 __all__ = ["interpolation_matrix", "prepare_interpolation_data", "interpolate_to_surface_submesh"]
 
@@ -333,7 +334,7 @@ def interpolate_to_surface_submesh(
     u_surface: dolfinx.fem.Function,
     submesh_facets: npt.NDArray[np.int32],
     integration_entities: npt.NDArray[np.int32],
-    entity_maps: list[dolfinx.mesh.EntityMap] | None = None,
+    entity_maps: list[_EntityMap] | None = None,
 ):
     """
     Interpolate a function `u_volume` into the function `u_surface`.

src/scifem/mesh.py

@@ -39,6 +39,7 @@ class _EntityMap(Protocol):
 
     sub_topology: dolfinx.mesh.Topology
     dim: int
+    _cpp_object: object
 
     def sub_topology_to_topology(
         self, entities: npt.NDArray[np.int32], inverse: bool