unilm/xtune/src/transformers/data/__init__.py

# flake8: noqa
# There's no way to ignore "F401 '...' imported but unused" warnings in this
# module, but to preserve other warnings. So, don't check this module at all.

from .metrics import is_sklearn_available
from .processors import (
    DataProcessor,
    InputExample,
    InputFeatures,
    SingleSentenceClassificationProcessor,
    SquadExample,
    SquadFeatures,
    SquadV1Processor,
    SquadV2Processor,
    glue_convert_examples_to_features,
    glue_output_modes,
    glue_processors,
    glue_tasks_num_labels,

    xglue_convert_examples_to_features,
    xglue_convert_examples_to_vat_features,
    xglue_output_modes,
    xglue_processors,
    xglue_tasks_num_labels,

    xtreme_convert_examples_to_features,
    xtreme_output_modes,
    xtreme_processors,
    xtreme_tasks_num_labels,

    squad_convert_examples_to_features,
    xnli_output_modes,
    xnli_processors,
    xnli_tasks_num_labels,
)


if is_sklearn_available():
    from .metrics import glue_compute_metrics, xnli_compute_metrics, xglue_compute_metrics, xtreme_compute_metrics