Source code for tensortrade.feed.core.mixins


from typing import List, Callable


[docs]class DataTypeMixin:
[docs] @classmethod def register_method(cls, func: "Callable", names: "List[str]"): """Injects methods into a specific stream instance. Parameters ---------- func : `Callable` The function to be injected as a method. names : `List[str]` The names to be given to the function. """ def method(self, *args, **kwargs): args = (self,) + args return func(*args, **kwargs) for name in names: setattr(cls, name, method)