Source code for tensortrade.base.registry


import numpy as np


_REGISTRY = {}


MAJOR_COMPONENTS = [
    'exchanges',
    'actions',
    'rewards',
    'features',
    'slippage'
]


[docs]def get_major_component_names():
    return MAJOR_COMPONENTS


[docs]def get_registry():
    return _REGISTRY


[docs]def registered_names():
    return list(np.unique([_REGISTRY[i] for i in _REGISTRY.keys()]))


[docs]def register(component, registered_name: str):
    _REGISTRY[component] = registered_name


[docs]def get_registered_name(component):
    return _REGISTRY[component]