NEDAS.models.lorenz96.tracer_advection.lorenz96_tracer_model module

class NEDAS.models.lorenz96.tracer_advection.lorenz96_tracer_model.Lorenz96TracerModel(**kwargs)[source]

Bases: Lorenz96Model

Lorenz 1996 model extended with a tracer advection component.

State vector is [lorenz_state (nx), tracer (nx), source (nx)].

mean_velocity: float

pert_velocity_multiplier: float

diffusion_coef: float

e_folding: float

sink_rate: float

point_tracer_source_rate: float

positive_tracer: bool

bound_above_is_one: bool

write_var_to_file(var, **kwargs)[source]

generate_initial_condition()[source]

run(*args, **kwargs)[source]

Run the model forward in time.

Parameters:

*args – Arguments
**kwargs – Keyword arguments

Keyword Arguments:

time (datetime) – current time when forecast starts
restart_dir (str) – directory where restart files are located
forecast_period (int) – forecast period in hours

If self.ens_run_strategy == ‘batch’, the method will run all ensemble members in one go, expect additional kwargs[‘nens’] to be the ensemble size. If self.ens_run_strategy == ‘scheduler’, the method runs a single member indexed by kwargs[‘member’], and kwargs[‘worker_id’] is the pid assigned by the scheduler to run this method.