Helper class for vectorized opt population calculation, via artificial states of length n_pop times n_states.

Attributes¶

states: foxes.core.States

The original states

n_pop: int

The population size

Public members¶

PopStates(states, n_pop)[source]

Constructor.

load_data(algo, verbosity=0)[source]

Load and/or create all model data that is subject to chunking.

initialize(algo, verbosity=0)[source]

Initializes the model.

size()[source]

The total number of states.

weights(algo)[source]

The statistical weights of all states.

output_point_vars(algo)[source]

The variables which are being modified by the model.

calculate(algo, mdata, fdata, pdata)[source]

” The main model calculation.

index()[source]

The index list

run_calculation(algo, *data, out_vars, **calc_pars)[source]

Starts the model calculation in parallel, via xarray’s apply_ufunc.

__repr__()[source]

Return repr(self).

property model_id

Unique id based on the model type.

var(v)[source]

Creates a model specific variable name.

property initialized

Initialization flag.

sub_models()[source]

List of all sub-models

finalize(algo, verbosity=0)[source]

Finalizes the model.

get_data(variable, target, lookup='smfp', mdata=None, ...)[source]

Getter for a data entry in the model object or provided data sources

data_to_store(name, algo, data)[source]

Adds data from mdata to the local store, intended for iterative runs.

from_data_or_store(name, algo, data, ret_dims=False, safe=False)[source]

Get data from mdata or local store