Source code for eegunity.modules.llm_booster.eeg_llm_booster

from eegunity.modules.llm_booster.eeg_llm_des_parser import llm_description_file_parser
from eegunity.modules.llm_booster.eeg_llm_file_parser import llm_boost_parser
from eegunity._share_attributes import _UDatasetSharedAttributes




class EEGLLMBooster(_UDatasetSharedAttributes):
    """
    This is a key module of `UnifiedDataset` class, with focus on large language boosting.
    This `EEGLLMBooster` class has the same attributes as the UnifiedDataset class. In this
    class, we define the functions relative to large language boosting.
    """
    def __init__(self, main_instance):
        super().__init__()
        self._shared_attr = main_instance._shared_attr
        self.llm_description_file_parser = llm_description_file_parser
        self.llm_boost_parser = llm_boost_parser