dymad.models.runtime_view

Narrow runtime-view adapter for model helper/component functions.

Functions

build_component_input_view(payload)

Build the narrow runtime-view adapter for model helpers.

Classes

ComponentInputView(runtime)

Expose runtime fields needed by model helpers through typed runtime views.

class dymad.models.runtime_view.ComponentInputView(runtime)

Bases: object

Expose runtime fields needed by model helpers through typed runtime views.

classmethod build(payload)
Return type:

ComponentInputView

property control: Tensor | None
property edge_attr: Tensor | None
property edge_index: Tensor
property edge_weight: Tensor | None
flatten_nodes(value)
Return type:

Tensor

property graph_control: Tensor | None
property graph_state: Tensor
runtime: TypeAliasType | TypeAliasType
property state: Tensor
unflatten_nodes(value)
Return type:

Tensor

dymad.models.runtime_view.build_component_input_view(payload)

Build the narrow runtime-view adapter for model helpers.

Return type:

ComponentInputView