spark.core.specs

spark.core.specs#

Classes#

`PortSpecs`	Base specification for a port of an SparkModule.
`PortMap`	Specification for an output port of an SparkModule.
`ModuleSpecs`	Specification for SparkModule automatic constructor.

Module Contents#

class spark.core.specs.PortSpecs(payload_type, shape, dtype, description=None, async_spikes=None, inhibition_mask=None)[source]#

Base specification for a port of an SparkModule.

Parameters:

payload_type (type[spark.core.payloads.SparkPayload] | None)
shape (tuple[int, Ellipsis] | list[tuple[int, Ellipsis]] | None)
dtype (jax.typing.DTypeLike | None)
description (str | None)
async_spikes (bool | None)
inhibition_mask (jax.Array | bool | None)

payload_type: type[spark.core.payloads.SparkPayload] | None[source]#

shape: tuple[int, Ellipsis] | list[tuple[int, Ellipsis]] | None[source]#

dtype: jax.typing.DTypeLike | None[source]#

description: str | None = None[source]#

async_spikes: bool | None = (None,)[source]#

inhibition_mask: bool | None = (None,)[source]#

to_dict(is_partial=False)[source]#

Serialize PortSpecs to dictionary

Parameters:: is_partial (bool)
Return type:: dict[str, Any]

classmethod from_dict(dct, is_partial=False)[source]#

Deserialize dictionary to PortSpecs

Parameters:

dct (dict)
is_partial (bool)

Return type:

Self

classmethod from_portspecs_list(portspec_list, validate_async=True)[source]#

Merges a list of PortSpecs into a single PortSpecs

Parameters:

portspec_list (list[PortSpecs])
validate_async (bool)

Return type:

Self

class spark.core.specs.PortMap(origin, port)[source]#

Specification for an output port of an SparkModule.

Parameters:

origin (str)
port (str)

origin: str[source]#

port: str[source]#

to_dict(is_partial=False)[source]#

Serialize PortMap to dictionary

Parameters:: is_partial (bool)
Return type:: dict[str, Any]

classmethod from_dict(dct, is_partial=False)[source]#

Deserialize dictionary to PortMap

Parameters:

dct (dict)
is_partial (bool)

Return type:

Self

class spark.core.specs.ModuleSpecs(name, module_cls, inputs, config)[source]#

Specification for SparkModule automatic constructor.

Parameters:

name (str)
module_cls (type)
inputs (dict[str, list[PortMap]])
config (spark.core.config.BaseSparkConfig)

name: str[source]#

module_cls: type[spark.core.module.SparkModule][source]#

inputs: dict[str, list[PortMap]][source]#

config: spark.core.config.BaseSparkConfig[source]#

to_dict(is_partial=False)[source]#

Serialize ModuleSpecs to dictionary

Parameters:: is_partial (bool)
Return type:: dict[str, Any]

classmethod from_dict(dct, is_partial=False)[source]#

Deserialize dictionary to ModuleSpecs

Parameters:

dct (dict)
is_partial (bool)

Return type:

Self