33
33
DASK_IMPORTED = False
34
34
35
35
36
- def forecast (vil , rainrate , velocity , n_timesteps , n_cascade_levels = 8 ,
36
+ def forecast (vil , velocity , n_timesteps , rainrate = None , n_cascade_levels = 8 ,
37
37
extrap_method = "semilagrangian" , ar_order = 2 , ar_window_radius = 50 ,
38
38
r_vil_window_radius = 5 , fft_method = "numpy" , num_workers = 1 ,
39
39
extrap_kwargs = None , filter_kwargs = None , measure_time = False ):
@@ -58,16 +58,16 @@ def forecast(vil, rainrate, velocity, n_timesteps, n_cascade_levels=8,
58
58
Array of shape (ar_order+2,m,n) containing the input fields ordered by
59
59
timestamp from oldest to newest. The time steps between the inputs are
60
60
assumed to be regular.
61
- rainrate : array_like
62
- Array of shape (m,n) containing the most recently observed rain rate
63
- field. If set to None, no R(VIL) conversion is done and the outputs
64
- are in the same units as the inputs.
65
61
velocity : array_like
66
62
Array of shape (2,m,n) containing the x- and y-components of the
67
63
advection field. The velocities are assumed to represent one time step
68
64
between the inputs. All values are required to be finite.
69
65
n_timesteps : int
70
66
Number of time steps to forecast.
67
+ rainrate : array_like
68
+ Array of shape (m,n) containing the most recently observed rain rate
69
+ field. If set to None, no R(VIL) conversion is done and the outputs
70
+ are in the same units as the inputs.
71
71
n_cascade_levels : int, optional
72
72
The number of cascade levels to use.
73
73
extrap_method : str, optional
0 commit comments