nessai.flowmodel.utils

Utilities for configuring FlowModel.

Module Contents

Functions

update_model_config(d)	Update the model (flow) configuration dictionary based on the defaults.
update_config(d)	Update the configuration dictionary to include the defaults.

nessai.flowmodel.utils.update_model_config(d)

Update the model (flow) configuration dictionary based on the defaults.

Parameters:

dUnion[dict, None]

Dictionary with the current configuration. If None, then the default is used.

Returns:

dict

Updated configuration dictionary.

Raises:

TypeError

Raised if the input is not a dictionary or None.

nessai.flowmodel.utils.update_config(d)

Update the configuration dictionary to include the defaults.

Parameters:

ddict

Dictionary with configuration

Returns:

dict

Dictionary with updated default configuration

Notes

The default configuration is specified in nessai.flowmodel.config

The kwargs can contain any additional keyword arguments that are specific to the type of flow being used.