Setting axis metadata for time

Author: erikvansebille

src/virtualship/expedition/input_data.py

@@ -105,6 +105,7 @@ def _load_ship_fieldset(cls, directory: Path) -> FieldSet:
         ds["bathymetry"] = -ds["bathymetry"]
         ds["depth"] = -ds["depth"]
         ds = ds.rename({"so": "S", "thetao": "T"})
+        ds.time.attrs["axis"] = "T"
         fieldset = FieldSet.from_copernicusmarine(ds)
         return fieldset
 
@@ -182,6 +183,7 @@ def _load_drifter_fieldset(cls, directory: Path) -> FieldSet:
         }
         ds = xr.open_mfdataset([filenames["U"], filenames["T"]])
         ds = ds.rename({"thetao": "T"})
+        ds.time.attrs["axis"] = "T"
         fieldset = FieldSet.from_copernicusmarine(ds)
         return fieldset