from small_text.integrations.pytorch.exceptions import PytorchNotFoundError

try:
    from small_text.integrations.transformers.datasets import TransformersDataset
    from small_text.integrations.transformers.classifiers.classification import (
        TransformerModelArguments,
        TransformerBasedClassification,
        TransformerBasedEmbeddingMixin)
except PytorchNotFoundError:
    pass

from small_text.integrations.pytorch.exceptions import PytorchNotFoundError

try:
    from small_text.integrations.transformers.datasets import TransformersDataset
    from small_text.integrations.transformers.classifiers.classification import (
        TransformerModelArguments,
        TransformerBasedClassification,
        TransformerBasedEmbeddingMixin)
except PytorchNotFoundError:
    pass
