gensbi.models.wrappers.conditional#

Classes#

ConditionalWrapper

Wrapper for conditional models to handle input expansion and calling convention.

Module Contents#

class gensbi.models.wrappers.conditional.ConditionalWrapper(model)[source]#

Bases: gensbi.utils.model_wrapping.ModelWrapper

Wrapper for conditional models to handle input expansion and calling convention.

Parameters:

model – The conditional model instance to wrap.

__call__(t, obs, obs_ids, cond, cond_ids, conditioned=True, guidance=None)[source]#

Call the wrapped model with expanded inputs.

Parameters:

  • t (Array) – Time steps.

  • obs (Array) – Observations.

  • obs_ids (Array) – Observation identifiers.

  • cond (Array) – Conditioning values.

  • cond_ids (Array) – Conditioning identifiers.

  • conditioned (bool | Array, optional) – Whether to use conditioning. Defaults to True.

  • guidance (Array | None, optional) – Optional guidance input.

Returns:

Model output.

Return type:

Array