bucky.model.npi#

Module to parse npi csv files.

Module Contents#

Functions#

get_npi_params(g_data, first_date, t_max, npi_file=None, disable_npi=False)

Read an npi scenario file or if none is provided provide correctly shaped 'no future changes' npi_params.

read_npi_file(fname, start_date, end_t, adm2_map, disable_npi=False)

TODO Description.
bucky.model.npi.get_npi_params(g_data, first_date, t_max, npi_file=None, disable_npi=False)[source]#

Read an npi scenario file or if none is provided provide correctly shaped ‘no future changes’ npi_params.

bucky.model.npi.read_npi_file(fname, start_date, end_t, adm2_map, disable_npi=False)[source]#

TODO Description.

Parameters:

  • fname (str) – Filename of NPI file

  • start_date (str) – Start date to use

  • end_t (int) – Number of days after start date

  • adm2_map (numpy.ndarray or cupy.ndarray if using CuPy) – Array of adm2 IDs

  • disable_npi (bool, optional) – Bool indicating whether NPIs should be disabled after being initialized for the first day

Returns:

npi_params – TODO

Return type:

dict