File size: 1,495 Bytes
079c32c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
from .registry import Registry
POLICY_REGISTRY = Registry()
ENV_REGISTRY = Registry()
ENV_WRAPPER_REGISTRY = Registry()
LEARNER_REGISTRY = Registry()
COMM_LEARNER_REGISTRY = Registry()
SERIAL_COLLECTOR_REGISTRY = Registry()
PARALLEL_COLLECTOR_REGISTRY = Registry()
COMM_COLLECTOR_REGISTRY = Registry()
BUFFER_REGISTRY = Registry()
COMMANDER_REGISTRY = Registry()
LEAGUE_REGISTRY = Registry()
PLAYER_REGISTRY = Registry()
MODEL_REGISTRY = Registry()
ENV_MANAGER_REGISTRY = Registry()
REWARD_MODEL_REGISTRY = Registry()
DATASET_REGISTRY = Registry()
SERIAL_EVALUATOR_REGISTRY = Registry()
MQ_REGISTRY = Registry()
WORLD_MODEL_REGISTRY = Registry()
STOCHASTIC_OPTIMIZER_REGISTRY = Registry()
registries = {
'policy': POLICY_REGISTRY,
'env': ENV_REGISTRY,
'env_wrapper': ENV_WRAPPER_REGISTRY,
'model': MODEL_REGISTRY,
'reward_model': REWARD_MODEL_REGISTRY,
'learner': LEARNER_REGISTRY,
'serial_collector': SERIAL_COLLECTOR_REGISTRY,
'parallel_collector': PARALLEL_COLLECTOR_REGISTRY,
'env_manager': ENV_MANAGER_REGISTRY,
'comm_learner': COMM_LEARNER_REGISTRY,
'comm_collector': COMM_COLLECTOR_REGISTRY,
'commander': COMMANDER_REGISTRY,
'league': LEAGUE_REGISTRY,
'player': PLAYER_REGISTRY,
'buffer': BUFFER_REGISTRY,
'dataset': DATASET_REGISTRY,
'serial_evaluator': SERIAL_EVALUATOR_REGISTRY,
'message_queue': MQ_REGISTRY,
'world_model': WORLD_MODEL_REGISTRY,
'stochastic_optimizer': STOCHASTIC_OPTIMIZER_REGISTRY,
}
|