Skip to content

Commit 46a1ed6

Browse files
Fixing CTD dataset ingestion
1 parent 44b724e commit 46a1ed6

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

src/virtualship/expedition/input_data.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -98,15 +98,16 @@ def _load_ship_fieldset(cls, directory: Path) -> FieldSet:
9898
"T": directory.joinpath("ship_t.nc"),
9999
"bathymetry": directory.joinpath("bathymetry.nc"),
100100
}
101-
ds = xr.open_mfdataset(
102-
[filenames["U"], filenames["T"], filenames["S"], filenames["bathymetry"]]
103-
)
104-
ds = ds.rename_vars({"deptho": "bathymetry"})
105-
ds["bathymetry"] = -ds["bathymetry"]
106-
ds["depth"] = -ds["depth"]
107-
ds = ds.reindex(depth=ds.depth[::-1])
108-
ds = ds.rename({"so": "S", "thetao": "T"})
109-
ds.time.attrs["axis"] = "T"
101+
dso = xr.open_mfdataset([filenames["U"], filenames["T"], filenames["S"]])
102+
dso["depth"] = -dso["depth"]
103+
dso = dso.reindex(depth=dso.depth[::-1])
104+
dso = dso.rename({"so": "S", "thetao": "T"})
105+
dso.time.attrs["axis"] = "T"
106+
107+
dsb = xr.open_dataset(filenames["bathymetry"]).rename_vars({"deptho": "bathymetry"})
108+
dsb["bathymetry"] = -dsb["bathymetry"]
109+
110+
ds = xr.merge([dso, dsb], join="inner")
110111
fieldset = FieldSet.from_copernicusmarine(ds)
111112
return fieldset
112113

0 commit comments

Comments
 (0)