From 36a89f08dde5efb83cf899b6870040128d078e13 Mon Sep 17 00:00:00 2001
From: Janice Kim <janice.kim@noaa.gov>
Date: Tue, 28 Oct 2025 10:50:11 -0400
Subject: [PATCH] Type tweak: Allow ZarrMonitor to take in Comm | DummyComm |
 None

---
 ndsl/monitor/zarr_monitor.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ndsl/monitor/zarr_monitor.py b/ndsl/monitor/zarr_monitor.py
index 99deec2b..1f57cf23 100644
--- a/ndsl/monitor/zarr_monitor.py
+++ b/ndsl/monitor/zarr_monitor.py
@@ -7,6 +7,7 @@
 import xarray as xr
 
 import ndsl.constants as constants
+from ndsl.comm import Comm
 from ndsl.comm.partitioner import Partitioner, subtile_slice
 from ndsl.logging import ndsl_log
 from ndsl.monitor.convert import to_numpy
@@ -47,7 +48,7 @@ def __init__(
         store: str | zarr.storage.MutableMapping,
         partitioner: Partitioner,
         mode: str = "w",
-        mpi_comm: DummyComm | None = None,
+        mpi_comm: Comm | DummyComm | None = None,
     ) -> None:
         """Create a ZarrMonitor.