dymad.models.rollout_engine

Typed rollout-engine selection for migrated model-spec families.

Functions

select_rollout_engine(model_spec, ...)

Resolve the rollout engine allowed by the typed model spec.

Classes

RolloutEngineSelection(predictor, source)

Resolved rollout engine for a typed model spec.

class dymad.models.rollout_engine.RolloutEngineSelection(predictor, source)

Bases: object

Resolved rollout engine for a typed model spec.

predictor: Callable

source: str

dymad.models.rollout_engine.select_rollout_engine(model_spec, model_config, dims)

Resolve the rollout engine allowed by the typed model spec.

Return type:

RolloutEngineSelection