Skip to content

Commit 42dd690

Browse files
Apply suggestions from code review
Co-authored-by: Erik van Sebille <e.vansebille@uu.nl>
1 parent a2620a6 commit 42dd690

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

src/parcels/_core/fieldset.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,13 @@ def from_ugrid_conventions(cls, ds: ux.UxDataset, mesh: str = "spherical"):
204204
"UV", fields["U"], fields["V"], vector_interp_method=Ux_Velocity
205205
)
206206
else:
207-
fields["UV"] = VectorField("UV", fields["U"], fields["V"], vector_interp_method=Ux_Velocity)
207+
fields["UV"] = VectorField("UV", fields["U"], fields["V"], vector_interp_method=Ux_Velocity)
208+
209+
if "W" in ds.data_vars:
210+
fields["W"] = Field("W", ds["W"], grid, _select_uxinterpolator(ds["W"]))
211+
fields["UVW"] = VectorField(
212+
"UVW", fields["U"], fields["V"], fields["W"], vector_interp_method=Ux_Velocity
213+
)
208214

209215
for varname in set(ds.data_vars) - set(fields.keys()):
210216
fields[varname] = Field(varname, ds[varname], grid, _select_uxinterpolator(ds[varname]))

0 commit comments

Comments
 (0)