10
10
from flopy .mbase import ModelInterface
11
11
from flopy .pakbase import PackageInterface
12
12
from flopy .plot .plotutil import PlotUtilities
13
- from xattree import _get_xatspec , _Xattribute
13
+ from xattree import Xattribute , _get_xatspec
14
14
15
15
from flopy4 .mf6 .model import Model
16
16
from flopy4 .mf6 .package import Package
@@ -197,6 +197,7 @@ def __init__(
197
197
for a in self ._data .attrs :
198
198
if (
199
199
self ._data .attrs [a ] is not None
200
+ and a in self ._spec .flat
200
201
and self ._spec .flat [a ].type is not None
201
202
):
202
203
d_fp3 = Flopy3Data (
@@ -211,7 +212,11 @@ def __init__(
211
212
self ._dlist .append (d_fp3 )
212
213
213
214
for v in self ._data .data_vars :
214
- if self ._data .data_vars [v ] is not None :
215
+ if (
216
+ self ._data .data_vars [v ] is not None
217
+ and v in self ._spec .flat
218
+ and self ._spec .flat [v ].type is not None
219
+ ):
215
220
d_fp3 = Flopy3Data (
216
221
data = self ._data .data_vars [v ],
217
222
spec = self ._spec .flat [v ],
@@ -276,7 +281,7 @@ class Flopy3Data(DataInterface):
276
281
def __init__ (
277
282
self ,
278
283
data ,
279
- spec : _Xattribute ,
284
+ spec : Xattribute ,
280
285
name : Optional [str ] = None ,
281
286
modelname : Optional [str ] = None ,
282
287
modelgrid : Optional [Grid ] = None ,
0 commit comments