clean fix_time

app.py

@@ -76,10 +76,10 @@ def greg_0h(jourjul):
     return gtime
 
 
-def fix_time(dataset):
+def fix_time(ds):
     from datetime import datetime, timedelta
 
-    time = dataset["TIME"]
+    time = ds["TIME"]
     time2 = time.dropna(dim="MAXT")
     time2 = time2.reindex_like(time, method="nearest")
     jourjul = [greg_0h(jourjul) for jourjul in time2.values]
@@ -87,11 +87,9 @@ def fix_time(dataset):
         datetime(jourjul[0], jourjul[1], jourjul[2], jourjul[3], jourjul[4], jourjul[5])
         for jourjul in jourjul
     ]
+    date = xr.DataArray(date, dims="MAXT")
+    return ds.assign(TIME=date)
 
-    return date
-
-
-from datetime import datetime, timedelta
 
 import cartopy.crs as ccrs
 import cartopy.feature as cfeature
@@ -302,9 +300,7 @@ class SADCP_Viewer(param.Parameterized):
                 ]
             ]
 
-            date = fix_time(self.ds)
-            date = xr.DataArray(date, dims="MAXT")
-            self.ds = self.ds.assign(TIME=date)
+            self.ds = fix_time(self.ds)
 
             lon_range = (
                 int(self.ds["LONGITUDE"].min().round() - 1),