fix type hints

Author: wpbonelli

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"),

flopy4/mf6/gwf/ic.py

@@ -1,3 +1,4 @@
+from typing import Optional
 import numpy as np
 from attrs import Converter
 from numpy.typing import NDArray
@@ -10,7 +11,7 @@
 
 @xattree
 class Ic(Package):
-    strt: NDArray[np.float64] = array(
+    strt: Optional[NDArray[np.float64]] = array(
         block="packagedata",
         dims=("nnodes",),
         default=1.0,