nessai.flowmodel.utils

Utilities for configuring FlowModel.

Functions

update_flow_config(cfg)	Update the model (flow) configuration dictionary based on the defaults.
update_config(flow_config[, training_config])	Update the configuration dictionary to include the defaults.

Module Contents

nessai.flowmodel.utils.update_flow_config(cfg)

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(flow_config, training_config=None)

Update the configuration dictionary to include the defaults.

Parameters:

flow_configdict

Dictionary with flow configuration

training_configdict

Dictionary with training config

Returns:

dict

Dictionary with updated flow configuration

dict

Dictionary with updated training 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.