Adopting imod open_cbc code

docs/examples/quickstart.py

@@ -57,4 +57,4 @@
 head.plot.imshow(ax=ax)
 head.plot.contour(ax=ax, levels=[0.2, 0.4, 0.6, 0.8], linewidths=3.0)
 budget.plot.quiver(x="x", y="y", u="npf-qx", v="npf-qy", ax=ax, color="white")
-fig.savefig(workspace / f"{name}.png")
+fig.savefig(workspace / ".." / "image" / "quickstart.png")

flopy4/mf6/gwf/__init__.py

@@ -2,7 +2,6 @@
 from typing import Optional
 
 import attrs
-import imod
 import xarray as xr
 from attrs import define
 from flopy.discretization.grid import Grid
@@ -15,7 +14,7 @@
 from flopy4.mf6.gwf.npf import Npf
 from flopy4.mf6.gwf.oc import Oc
 from flopy4.mf6.model import Model
-from flopy4.mf6.utils import open_hds
+from flopy4.mf6.utils import open_cbc, open_hds
 
 __all__ = ["Gwf", "Chd", "Dis", "Ic", "Npf", "Oc"]
 
@@ -45,10 +44,9 @@ def head(self) -> xr.DataArray:
         @property
         def budget(self):
             # TODO support other extensions than .bud (e.g. .cbc)
-            return imod.mf6.open_cbc(
+            return open_cbc(
                 self.parent.parent.path / f"{self.parent.name}.bud",
                 self.parent.parent.path / f"{self.parent.name}.dis.grb",
-                merge_to_dataset=True,
             )
 
     dis: Dis = field(converter=convert_grid)

flopy4/mf6/utils/__init__.py

@@ -1,3 +1,4 @@
+from .cbc_reader import open_cbc
 from .heads_reader import open_hds
 
-__all__ = ["open_hds"]
+__all__ = ["open_hds", "open_cbc"]