from torch import nn
from configs.config import ConfigManager


class Pipeline(nn.Module):
    TRAIN_MODE: str = "train"
    TEST_MODE: str = "test"

    def __init__(self, pipeline_config):
        super(Pipeline, self).__init__()

        self.modules_config = ConfigManager.get("modules")
        self.device = ConfigManager.get("settings", "general", "device")

    def forward(self, data, mode):
        pass


if __name__ == '__main__':
    pass