Commit 2e750c53 authored by Clément Haëck's avatar Clément Haëck
Browse files

Allow multiple datasets for fix_time_daily

parent 85e1545f
......@@ -169,7 +169,10 @@ def progressbar(it, prefix="", size=60, file=sys.stdout):
file.flush()
def fix_time_daily(ds):
ds = ds.assign_coords(
time=ds.time.dt.floor("D"))
return ds
def fix_time_daily(*datasets):
out = []
for ds in datasets:
out.append(ds.assign_coords(time=ds.time.dt.floor("D")))
if len(out) == 1:
return out[0]
return out
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment