replay_trajectory_classification.initial_conditions.UniformOneEnvironmentInitialConditions#

class UniformOneEnvironmentInitialConditions(environment_name: str = '')[source]#

Bases: object

Initial conditions where all position bins are equally likely for one environment and zero for other environments.

Methods

make_initial_conditions(environments, ...)

Creates initial conditions array

Methods

make_initial_conditions

Creates initial conditions array

Attributes

environment_name

make_initial_conditions(environments: tuple[Environment], environment_names_to_state: tuple[str]) → list[ndarray][source]#

Creates initial conditions array

Parameters:

  • environments (tuple[Environment]) – Spatial environments in the model

  • environment_names_to_state (tuple[str]) – Mapping of environment names to state

Returns:

initial_conditions

Return type:

list of arrays