Source code for catalyst.rl.core.db

from abc import ABC, abstractmethod
from enum import Enum


class DBSpec(ABC):
    class Message(Enum):
        ENABLE_TRAINING = 0
        DISABLE_TRAINING = 1
        ENABLE_SAMPLING = 2
        DISABLE_SAMPLING = 3

    @property
    @abstractmethod
    def training_enabled(self) -> bool:
        pass

    @property
    @abstractmethod
    def sampling_enabled(self) -> bool:
        pass

    @property
    @abstractmethod
    def epoch(self) -> int:
        pass

    @property
    @abstractmethod
    def num_trajectories(self) -> int:
        pass

    @abstractmethod
    def push_message(self, message: Message):
        pass

    @abstractmethod
    def put_trajectory(self, trajectory, raw: bool):
        pass

    @abstractmethod
    def get_trajectory(self, index=None):
        pass

    @abstractmethod
    def del_trajectory(self):
        pass

    @abstractmethod
    def put_checkpoint(self, checkpoint, epoch):
        pass

    @abstractmethod
    def get_checkpoint(self):
        pass

    @abstractmethod
    def del_checkpoint(self):
        pass