mypy fix?

illinois-ceesd · Aug 28, 2024 · 0b491e9 · 0b491e9
1 parent 16a3a6a
commit 0b491e9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mirgecom/array_context.py b/mirgecom/array_context.py
@@ -243,6 +243,8 @@ def log_disk_cache_config(actx: ArrayContext) -> None:
     assert isinstance(actx, (PyOpenCLArrayContext, PytatoPyOpenCLArrayContext))
 
     if actx_class_is_distributed(type(actx)):
+        from grudge.array_context import MPIBasedArrayContext
+        assert isinstance(actx, MPIBasedArrayContext)
         rank = actx.mpi_communicator.Get_rank()
     else:
         rank = 0