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