Novel Respiratory (SIR)

An SIR compartmental model for novel respiratory diseases with age-stratified transmission

CovidSIRModel(config)

Bases: CovidJaxModel

Source code in compartment/models/covid_jax_model/model.py

def __init__(self, config):
    """Initialize the model and adopt the config's compartment list.

    The config's ``compartment_list`` may be a variant subset, so it
    replaces the full schema-derived list.

    Args:
        config (dict): Validated simulation configuration, including
            ``compartment_list``.
    """
    super().__init__(config)

    # Override with the compartment list from config (may be a variant subset).
    self.compartment_list = config["compartment_list"]