write some tests and stick grid and time in model and sim respectively as attrs. maybe the way to go? but we need to repeat them in all relevant packages in that case. we could maybe also do a decorator that adds grid/time via field_transformer? but that's probably too fancy

Author: wpbonelli

flopy4/mf6/gwf/__init__.py

@@ -1,7 +1,9 @@
 from pathlib import Path
 from typing import Optional
 
+import attrs
 from attrs import define
+from flopy.discretization.grid import Grid
 from xattree import field, xattree
 
 from flopy4.mf6.gwf.chd import Chd
@@ -21,6 +23,7 @@ class Gwf(Model):
     oc: Oc = field()
     npf: Npf = field()
     chd: list[Chd] = field()
+    grid: Grid = attrs.field(default=None)
 
     @define
     class NewtonOptions:

flopy4/mf6/simulation.py

@@ -1,3 +1,5 @@
+import attrs
+from flopy.discretization.modeltime import ModelTime
 from xattree import field, xattree
 
 from flopy4.mf6.component import Component
@@ -13,3 +15,4 @@ class Simulation(Component):
     exchanges: dict[str, Exchange] = field()
     solutions: dict[str, Solution] = field()
     tdis: Tdis = field()
+    time: ModelTime = attrs.field(default=None)