"""`smiles_gpt` implements transformer models for molecule generation and molecular-
property prediction.
"""

__author__ = "Sanjar Ad[iy]lov"
__version__ = "1.0.0-pub"

from . import classification, data, language_modeling, tokenization
from .classification import (ClassifierLitModel, RegressorLitModel,
                             GPT2ForSequenceClassification)
from .data import CSVDataModule, CVSplitter, LMDataModule
from .language_modeling import GPT2LitModel
from .tokenization import SMILESBPETokenizer, SMILESAlphabet
from .iupac_tokenization_iupac import get_data_loader,prepare_input
from .iupac_tokenization_pro import get_data_loader_pro,prepare_input_pro
from .iupac_tokenization_class import get_data_loader_class,prepare_input_class

__all__ = ("classification", "data", "tokenization",
           "ClassifierLitModel", "CSVDataModule", "CVSplitter",
           "GPT2ForSequenceClassification", "GPT2LitModel", "LMDataModule",
           "RegressorLitModel", "SMILESBPETokenizer", "SMILESAlphabet")