expanding only lat

CMCC-Foundation · Jul 17, 2024 · 5c4cea8 · 5c4cea8
1 parent 15772cd
commit 5c4cea8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/drivers/intake_geokube/afm.py b/drivers/intake_geokube/afm.py
@@ -17,8 +17,8 @@ def postprocess_afm(dset: xr.Dataset) -> xr.Dataset:
     longitude = dset['lon'].values
     dset = dset.drop('lat')
     dset = dset.drop('lon')
-    #return dset.expand_dims(dim={"lat": latitude, "lon": longitude}, axis=(1, 0)).sortby('time')
-    return dset.sortby('time')
+    dset = dset.sortby('time')
+    return dset.expand_dims(dim={"lat": latitude}, axis=0) #, "lon": longitude}, axis=(1, 0))
 
 
 class CMCCAFMSource(GeokubeSource):