Source code for vein.sources.memory.callable

from typing import Callable

from vein.flow.state import AbstractLiveStateTracker



[docs]
class CallableTracker(AbstractLiveStateTracker):

    def __init__(self, callable: Callable, *args, **kwargs) -> None:
        super().__init__(**kwargs)
        self._callable = callable

        self._args = args
        self._kwargs = kwargs


[docs]
    def get_state_value(self):
        return self._callable(*self._args, **self._kwargs)