STY: Try to make mypy happy (again)

src/porepy/applications/md_grids/model_geometries.py

@@ -1,8 +1,11 @@
 from __future__ import annotations
 
+from typing import Union, cast
+
 import numpy as np
 
 import porepy as pp
+from porepy.grids import mortar_grid
 from porepy.grids.mdg_generation import _preprocess_cartesian_args
 
 from . import domains, fracture_sets
@@ -255,7 +258,15 @@ def set_geometry(self) -> None:
 
         # Finally replace the subdomains and interfaces in the original
         # mixed-dimensional grid. Both can be done at the same time:
-        self.mdg.replace_subdomains_and_interfaces(sd_map=grid_map, intf_map=intf_map)
-
+        self.mdg.replace_subdomains_and_interfaces(
+            sd_map=grid_map,
+            intf_map=cast(
+                dict[
+                    pp.MortarGrid,
+                    Union[pp.MortarGrid, dict[mortar_grid.MortarSides, pp.Grid]],
+                ],
+                intf_map,
+            ),
+        )
         # Create projections between local and global coordinates for fracture grids.
         pp.set_local_coordinate_projections(self.mdg)

src/porepy/grids/md_grid.py

@@ -5,7 +5,6 @@
 
 from __future__ import annotations
 
-from collections.abc import Mapping
 from typing import Any, Callable, Iterable, Literal, Optional, Union, overload
 
 import numpy as np
@@ -698,11 +697,11 @@ def copy(self) -> MixedDimensionalGrid:
 
     def replace_subdomains_and_interfaces(
         self,
-        sd_map: Optional[Mapping[pp.Grid, pp.Grid]] = None,
+        sd_map: Optional[dict[pp.Grid, pp.Grid]] = None,
         intf_map: Optional[
-            Mapping[
+            dict[
                 pp.MortarGrid,
-                Union[pp.MortarGrid, Mapping[mortar_grid.MortarSides, pp.Grid]],
+                Union[pp.MortarGrid, dict[mortar_grid.MortarSides, pp.Grid]],
             ]
         ] = None,
         tol: float = 1e-6,