list_registered_behaviors

list_registered_behaviors#

class brainstate.environ.list_registered_behaviors(*, env=None)[source]#

List all keys with registered callbacks.

Parameters:

env (EnvironmentState | None) – The environment state to query. If None, uses the global environment.

Returns:

Keys that have registered behavior callbacks.

Return type:

List[str]

Examples

>>> import brainstate.environ as env
>>>
>>> # Register some callbacks
>>> env.register_default_behavior('param1', lambda x: None)
>>> env.register_default_behavior('param2', lambda x: None)
>>>
>>> # List registered behaviors
>>> behaviors = env.list_registered_behaviors()
>>> print(f"Registered: {behaviors}")  # ['param1', 'param2']
>>>
>>> # Check if specific behavior is registered
>>> if 'dt' in behaviors:
...     print("dt has a registered callback")

Using custom environment:

>>> import brainstate.environ as env
>>>
>>> custom_env = env.EnvironmentState()
>>> env.register_default_behavior('param', lambda x: None, env=custom_env)
>>> print(env.list_registered_behaviors(env=custom_env))  # ['param']