breaking the expand dim operation into two steps

CMCC-Foundation · Jul 17, 2024 · 0402069 · 0402069
1 parent 25d07d1
commit 0402069
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/drivers/intake_geokube/afm.py b/drivers/intake_geokube/afm.py
@@ -18,7 +18,9 @@ def postprocess_afm(dset: xr.Dataset) -> xr.Dataset:
     dset = dset.drop('lat')
     dset = dset.drop('lon')
     dset = dset.sortby('time')
-    return dset.expand_dims(dim={"lat": latitude}, axis=0).expand_dims(dim={"lon": longitude}, axis=0)
+    dset = dset.expand_dims(dim={"lat": latitude}, axis=0)
+    dset = dset.expand_dims(dim={"lon": longitude}, axis=0)
+    return dset
 
 
 class CMCCAFMSource(GeokubeSource):