Index

_

__init__() (aind_behavior_curriculum.base.AindBehaviorModel method) (aind_behavior_curriculum.base.AindBehaviorModelExtra method) (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.Curriculum method) (aind_behavior_curriculum.curriculum.Metrics method) (aind_behavior_curriculum.curriculum.MetricsProvider method) (aind_behavior_curriculum.curriculum.Policy method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.PolicyTransition method) (aind_behavior_curriculum.curriculum.Stage method) (aind_behavior_curriculum.curriculum.StageGraph method) (aind_behavior_curriculum.curriculum.StageTransition method) (aind_behavior_curriculum.curriculum_utils.Graduated method) (aind_behavior_curriculum.task.Task method) (aind_behavior_curriculum.task.TaskParameters method) (aind_behavior_curriculum.trainer.Trainer method) (aind_behavior_curriculum.trainer.TrainerServer method) (aind_behavior_curriculum.trainer.TrainerState method)

A

add_node() (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.StageGraph method) add_policy() (aind_behavior_curriculum.curriculum.Stage method) add_policy_transition() (aind_behavior_curriculum.curriculum.Stage method) add_stage() (aind_behavior_curriculum.curriculum.Curriculum method) add_stage_transition() (aind_behavior_curriculum.curriculum.Curriculum method) add_transition() (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.StageGraph method) aind_behavior_curriculum module

aind_behavior_curriculum.base module aind_behavior_curriculum.curriculum module aind_behavior_curriculum.curriculum_utils module aind_behavior_curriculum.task module aind_behavior_curriculum.trainer module AindBehaviorModel (class in aind_behavior_curriculum.base) AindBehaviorModelExtra (class in aind_behavior_curriculum.base)

B

BehaviorGraph (class in aind_behavior_curriculum.curriculum)

C

callable (aind_behavior_curriculum.curriculum.MetricsProvider property) (aind_behavior_curriculum.curriculum.Policy property) (aind_behavior_curriculum.curriculum.PolicyTransition property) (aind_behavior_curriculum.curriculum.StageTransition property) copy() (aind_behavior_curriculum.base.AindBehaviorModel method) (aind_behavior_curriculum.base.AindBehaviorModelExtra method) (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.Curriculum method) (aind_behavior_curriculum.curriculum.Metrics method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.Stage method) (aind_behavior_curriculum.curriculum.StageGraph method) (aind_behavior_curriculum.curriculum_utils.Graduated method) (aind_behavior_curriculum.task.Task method) (aind_behavior_curriculum.task.TaskParameters method) (aind_behavior_curriculum.trainer.TrainerState method)

create_curriculum() (in module aind_behavior_curriculum.curriculum) create_task() (in module aind_behavior_curriculum.task) create_trainer_state() (aind_behavior_curriculum.trainer.Trainer method) curriculum (aind_behavior_curriculum.trainer.Trainer property) Curriculum (class in aind_behavior_curriculum.curriculum) curriculum() (aind_behavior_curriculum.trainer.Trainer method)

D

default() (aind_behavior_curriculum.trainer.TrainerState class method)

default_pkg_location_factory() (aind_behavior_curriculum.curriculum.Curriculum class method)

E

eject_subject() (aind_behavior_curriculum.trainer.TrainerServer method) evaluate() (aind_behavior_curriculum.trainer.Trainer method)

evaluate_subjects() (aind_behavior_curriculum.trainer.TrainerServer method) export_diagram() (in module aind_behavior_curriculum.curriculum_utils) export_json() (in module aind_behavior_curriculum.curriculum_utils)

G

get_net_parameter_update() (aind_behavior_curriculum.trainer.Trainer static method)

get_task_parameters() (aind_behavior_curriculum.curriculum.Stage method) Graduated (class in aind_behavior_curriculum.curriculum_utils)

I

invoke() (aind_behavior_curriculum.curriculum.MetricsProvider method) (aind_behavior_curriculum.curriculum.Policy method) (aind_behavior_curriculum.curriculum.PolicyTransition method) (aind_behavior_curriculum.curriculum.StageTransition method)

is_non_deserializable_callable() (in module aind_behavior_curriculum.curriculum)

L

load_data() (aind_behavior_curriculum.trainer.TrainerServer method)

M

make_task_discriminator() (in module aind_behavior_curriculum.curriculum) Metrics (class in aind_behavior_curriculum.curriculum) MetricsProvider (class in aind_behavior_curriculum.curriculum) model_config (aind_behavior_curriculum.base.AindBehaviorModel attribute) (aind_behavior_curriculum.base.AindBehaviorModelExtra attribute) (aind_behavior_curriculum.curriculum.BehaviorGraph attribute) (aind_behavior_curriculum.curriculum.Curriculum attribute) (aind_behavior_curriculum.curriculum.Metrics attribute) (aind_behavior_curriculum.curriculum.PolicyGraph attribute) (aind_behavior_curriculum.curriculum.Stage attribute) (aind_behavior_curriculum.curriculum.StageGraph attribute) (aind_behavior_curriculum.curriculum_utils.Graduated attribute) (aind_behavior_curriculum.task.Task attribute) (aind_behavior_curriculum.task.TaskParameters attribute) (aind_behavior_curriculum.trainer.TrainerState attribute) model_construct() (aind_behavior_curriculum.base.AindBehaviorModel class method) (aind_behavior_curriculum.base.AindBehaviorModelExtra class method) (aind_behavior_curriculum.curriculum.BehaviorGraph class method) (aind_behavior_curriculum.curriculum.Curriculum class method) (aind_behavior_curriculum.curriculum.Metrics class method) (aind_behavior_curriculum.curriculum.PolicyGraph class method) (aind_behavior_curriculum.curriculum.Stage class method) (aind_behavior_curriculum.curriculum.StageGraph class method) (aind_behavior_curriculum.curriculum_utils.Graduated class method) (aind_behavior_curriculum.task.Task class method) (aind_behavior_curriculum.task.TaskParameters class method) (aind_behavior_curriculum.trainer.TrainerState class method) model_copy() (aind_behavior_curriculum.base.AindBehaviorModel method) (aind_behavior_curriculum.base.AindBehaviorModelExtra method) (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.Curriculum method) (aind_behavior_curriculum.curriculum.Metrics method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.Stage method) (aind_behavior_curriculum.curriculum.StageGraph method) (aind_behavior_curriculum.curriculum_utils.Graduated method) (aind_behavior_curriculum.task.Task method) (aind_behavior_curriculum.task.TaskParameters method) (aind_behavior_curriculum.trainer.TrainerState method) model_dump() (aind_behavior_curriculum.base.AindBehaviorModel method) (aind_behavior_curriculum.base.AindBehaviorModelExtra method) (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.Curriculum method) (aind_behavior_curriculum.curriculum.Metrics method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.Stage method) (aind_behavior_curriculum.curriculum.StageGraph method) (aind_behavior_curriculum.curriculum_utils.Graduated method) (aind_behavior_curriculum.task.Task method) (aind_behavior_curriculum.task.TaskParameters method) (aind_behavior_curriculum.trainer.TrainerState method) model_dump_json() (aind_behavior_curriculum.base.AindBehaviorModel method) (aind_behavior_curriculum.base.AindBehaviorModelExtra method) (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.Curriculum method) (aind_behavior_curriculum.curriculum.Metrics method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.Stage method) (aind_behavior_curriculum.curriculum.StageGraph method) (aind_behavior_curriculum.curriculum_utils.Graduated method) (aind_behavior_curriculum.task.Task method) (aind_behavior_curriculum.task.TaskParameters method) (aind_behavior_curriculum.trainer.TrainerState method) model_extra (aind_behavior_curriculum.base.AindBehaviorModel property) (aind_behavior_curriculum.base.AindBehaviorModelExtra property) (aind_behavior_curriculum.curriculum.BehaviorGraph property) (aind_behavior_curriculum.curriculum.Curriculum property) (aind_behavior_curriculum.curriculum.Metrics property) (aind_behavior_curriculum.curriculum.PolicyGraph property) (aind_behavior_curriculum.curriculum.Stage property) (aind_behavior_curriculum.curriculum.StageGraph property) (aind_behavior_curriculum.curriculum_utils.Graduated property) (aind_behavior_curriculum.task.Task property) (aind_behavior_curriculum.task.TaskParameters property) (aind_behavior_curriculum.trainer.TrainerState property) model_fields_set (aind_behavior_curriculum.base.AindBehaviorModel property) (aind_behavior_curriculum.base.AindBehaviorModelExtra property) (aind_behavior_curriculum.curriculum.BehaviorGraph property) (aind_behavior_curriculum.curriculum.Curriculum property) (aind_behavior_curriculum.curriculum.Metrics property) (aind_behavior_curriculum.curriculum.PolicyGraph property) (aind_behavior_curriculum.curriculum.Stage property) (aind_behavior_curriculum.curriculum.StageGraph property) (aind_behavior_curriculum.curriculum_utils.Graduated property) (aind_behavior_curriculum.task.Task property) (aind_behavior_curriculum.task.TaskParameters property) (aind_behavior_curriculum.trainer.TrainerState property) model_json_schema() (aind_behavior_curriculum.base.AindBehaviorModel class method) (aind_behavior_curriculum.base.AindBehaviorModelExtra class method) (aind_behavior_curriculum.curriculum.BehaviorGraph class method) (aind_behavior_curriculum.curriculum.Curriculum class method) (aind_behavior_curriculum.curriculum.Metrics class method) (aind_behavior_curriculum.curriculum.PolicyGraph class method) (aind_behavior_curriculum.curriculum.Stage class method) (aind_behavior_curriculum.curriculum.StageGraph class method) (aind_behavior_curriculum.curriculum_utils.Graduated class method) (aind_behavior_curriculum.task.Task class method) (aind_behavior_curriculum.task.TaskParameters class method) (aind_behavior_curriculum.trainer.TrainerState class method)

model_parametrized_name() (aind_behavior_curriculum.base.AindBehaviorModel class method) (aind_behavior_curriculum.base.AindBehaviorModelExtra class method) (aind_behavior_curriculum.curriculum.BehaviorGraph class method) (aind_behavior_curriculum.curriculum.Curriculum class method) (aind_behavior_curriculum.curriculum.Metrics class method) (aind_behavior_curriculum.curriculum.PolicyGraph class method) (aind_behavior_curriculum.curriculum.Stage class method) (aind_behavior_curriculum.curriculum.StageGraph class method) (aind_behavior_curriculum.curriculum_utils.Graduated class method) (aind_behavior_curriculum.task.Task class method) (aind_behavior_curriculum.task.TaskParameters class method) (aind_behavior_curriculum.trainer.TrainerState class method) model_post_init() (aind_behavior_curriculum.base.AindBehaviorModel method) (aind_behavior_curriculum.base.AindBehaviorModelExtra method) (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.Curriculum method) (aind_behavior_curriculum.curriculum.Metrics method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.Stage method) (aind_behavior_curriculum.curriculum.StageGraph method) (aind_behavior_curriculum.curriculum_utils.Graduated method) (aind_behavior_curriculum.task.Task method) (aind_behavior_curriculum.task.TaskParameters method) (aind_behavior_curriculum.trainer.TrainerState method) model_rebuild() (aind_behavior_curriculum.base.AindBehaviorModel class method) (aind_behavior_curriculum.base.AindBehaviorModelExtra class method) (aind_behavior_curriculum.curriculum.BehaviorGraph class method) (aind_behavior_curriculum.curriculum.Curriculum class method) (aind_behavior_curriculum.curriculum.Metrics class method) (aind_behavior_curriculum.curriculum.PolicyGraph class method) (aind_behavior_curriculum.curriculum.Stage class method) (aind_behavior_curriculum.curriculum.StageGraph class method) (aind_behavior_curriculum.curriculum_utils.Graduated class method) (aind_behavior_curriculum.task.Task class method) (aind_behavior_curriculum.task.TaskParameters class method) (aind_behavior_curriculum.trainer.TrainerState class method) model_validate() (aind_behavior_curriculum.base.AindBehaviorModel class method) (aind_behavior_curriculum.base.AindBehaviorModelExtra class method) (aind_behavior_curriculum.curriculum.BehaviorGraph class method) (aind_behavior_curriculum.curriculum.Curriculum class method) (aind_behavior_curriculum.curriculum.Metrics class method) (aind_behavior_curriculum.curriculum.PolicyGraph class method) (aind_behavior_curriculum.curriculum.Stage class method) (aind_behavior_curriculum.curriculum.StageGraph class method) (aind_behavior_curriculum.curriculum_utils.Graduated class method) (aind_behavior_curriculum.task.Task class method) (aind_behavior_curriculum.task.TaskParameters class method) (aind_behavior_curriculum.trainer.TrainerState class method) model_validate_json() (aind_behavior_curriculum.base.AindBehaviorModel class method) (aind_behavior_curriculum.base.AindBehaviorModelExtra class method) (aind_behavior_curriculum.curriculum.BehaviorGraph class method) (aind_behavior_curriculum.curriculum.Curriculum class method) (aind_behavior_curriculum.curriculum.Metrics class method) (aind_behavior_curriculum.curriculum.PolicyGraph class method) (aind_behavior_curriculum.curriculum.Stage class method) (aind_behavior_curriculum.curriculum.StageGraph class method) (aind_behavior_curriculum.curriculum_utils.Graduated class method) (aind_behavior_curriculum.task.Task class method) (aind_behavior_curriculum.task.TaskParameters class method) (aind_behavior_curriculum.trainer.TrainerState class method) model_validate_strings() (aind_behavior_curriculum.base.AindBehaviorModel class method) (aind_behavior_curriculum.base.AindBehaviorModelExtra class method) (aind_behavior_curriculum.curriculum.BehaviorGraph class method) (aind_behavior_curriculum.curriculum.Curriculum class method) (aind_behavior_curriculum.curriculum.Metrics class method) (aind_behavior_curriculum.curriculum.PolicyGraph class method) (aind_behavior_curriculum.curriculum.Stage class method) (aind_behavior_curriculum.curriculum.StageGraph class method) (aind_behavior_curriculum.curriculum_utils.Graduated class method) (aind_behavior_curriculum.task.Task class method) (aind_behavior_curriculum.task.TaskParameters class method) (aind_behavior_curriculum.trainer.TrainerState class method) module aind_behavior_curriculum aind_behavior_curriculum.base aind_behavior_curriculum.curriculum aind_behavior_curriculum.curriculum_utils aind_behavior_curriculum.task aind_behavior_curriculum.trainer

N

name (aind_behavior_curriculum.curriculum.MetricsProvider property) (aind_behavior_curriculum.curriculum.Policy property) (aind_behavior_curriculum.curriculum.PolicyTransition property) (aind_behavior_curriculum.curriculum.StageTransition property)

normalize_rule_or_callable() (aind_behavior_curriculum.curriculum.MetricsProvider class method) (aind_behavior_curriculum.curriculum.Policy class method) (aind_behavior_curriculum.curriculum.PolicyTransition class method) (aind_behavior_curriculum.curriculum.StageTransition class method)

O

override_subject_status() (aind_behavior_curriculum.trainer.TrainerServer method)

P

Policy (class in aind_behavior_curriculum.curriculum)

PolicyGraph (class in aind_behavior_curriculum.curriculum) PolicyTransition (class in aind_behavior_curriculum.curriculum)

R

register_subject() (aind_behavior_curriculum.trainer.TrainerServer method) remove_node() (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.StageGraph method) remove_node_transition() (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.StageGraph method)

remove_policy() (aind_behavior_curriculum.curriculum.Stage method) remove_policy_transition() (aind_behavior_curriculum.curriculum.Stage method) remove_stage() (aind_behavior_curriculum.curriculum.Curriculum method) remove_stage_transition() (aind_behavior_curriculum.curriculum.Curriculum method)

S

see_node_transitions() (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.StageGraph method) see_nodes() (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.StageGraph method) see_policies() (aind_behavior_curriculum.curriculum.Stage method) see_policy_transitions() (aind_behavior_curriculum.curriculum.Stage method) see_stage_transitions() (aind_behavior_curriculum.curriculum.Curriculum method) see_stages() (aind_behavior_curriculum.curriculum.Curriculum method) serialize_rule() (aind_behavior_curriculum.curriculum.MetricsProvider class method) (aind_behavior_curriculum.curriculum.Policy class method) (aind_behavior_curriculum.curriculum.PolicyTransition class method) (aind_behavior_curriculum.curriculum.StageTransition class method)

set_policy_transition_priority() (aind_behavior_curriculum.curriculum.Stage method) set_stage_transition_priority() (aind_behavior_curriculum.curriculum.Curriculum method) set_start_policies() (aind_behavior_curriculum.curriculum.Stage method) set_task_parameters() (aind_behavior_curriculum.curriculum.Stage method) set_transition_priority() (aind_behavior_curriculum.curriculum.BehaviorGraph method) (aind_behavior_curriculum.curriculum.PolicyGraph method) (aind_behavior_curriculum.curriculum.StageGraph method) Stage (class in aind_behavior_curriculum.curriculum) StageGraph (class in aind_behavior_curriculum.curriculum) StageTransition (class in aind_behavior_curriculum.curriculum)

T

Task (class in aind_behavior_curriculum.task) task_discriminator_type() (aind_behavior_curriculum.curriculum.Curriculum method) TaskParameters (class in aind_behavior_curriculum.task)

Trainer (class in aind_behavior_curriculum.trainer) TrainerServer (class in aind_behavior_curriculum.trainer) TrainerState (class in aind_behavior_curriculum.trainer) try_materialize_non_deserializable_callable_error() (in module aind_behavior_curriculum.curriculum)

V

validate_curriculum() (aind_behavior_curriculum.curriculum.Curriculum method)

validate_stage() (aind_behavior_curriculum.curriculum.Stage method)

W

write_data() (aind_behavior_curriculum.trainer.TrainerServer method)