Commit cfe4af4c authored by tpys's avatar tpys
Browse files

add split param for save_like

parent 6dd8498d
......@@ -25,10 +25,15 @@ def split_variable(ds, name):
return v
def save_like(output, input, step, save_dir="", freq=6, split=False):
def save_like(output, input, step, input_type="hres", save_dir="", freq=6, split=False):
if save_dir:
os.makedirs(save_dir, exist_ok=True)
dtime = (step+2) * freq #
if input_type == "hres":
dtime = (step+2) * freq
elif input_type == "gfs":
dtime = (step+1) * freq
init_time = pd.to_datetime(input.time.values[0])
ds = xr.DataArray(
......
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