From 17505b3b46b606f27586e704eb5d212c3f3627bd Mon Sep 17 00:00:00 2001
From: wpbonelli <wpbonelli@gmail.com>
Date: Tue, 8 Jul 2025 12:46:11 -0400
Subject: [PATCH] fix type hints

---
 flopy4/mf6/gwf/dis.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/flopy4/mf6/gwf/dis.py b/flopy4/mf6/gwf/dis.py
index 174188b3..e069e31f 100644
--- a/flopy4/mf6/gwf/dis.py
+++ b/flopy4/mf6/gwf/dis.py
@@ -1,3 +1,4 @@
+from typing import Optional
 import numpy as np
 from attrs import Converter
 from flopy.discretization.structuredgrid import StructuredGrid
@@ -65,7 +66,7 @@ class Dis(Package):
         dims=("nlay", "nrow", "ncol"),
         converter=Converter(dict_to_array, takes_self=True, takes_field=True),
     )
-    idomain: NDArray[np.int32] = array(
+    idomain: Optional[NDArray[np.int32]] = array(
         block="griddata",
         default=1,
         dims=("nlay", "nrow", "ncol"),