




from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_tokenizers_available, is_torch_available


_UpperCAmelCase						:    List[str]      =  {
    "configuration_markuplm": ["MARKUPLM_PRETRAINED_CONFIG_ARCHIVE_MAP", "MarkupLMConfig"],
    "feature_extraction_markuplm": ["MarkupLMFeatureExtractor"],
    "processing_markuplm": ["MarkupLMProcessor"],
    "tokenization_markuplm": ["MarkupLMTokenizer"],
}

try:
     if not is_tokenizers_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     _UpperCAmelCase						:    List[str]      =  ["MarkupLMTokenizerFast"]

try:
     if not is_torch_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     _UpperCAmelCase						:    Optional[Any]      =  [
         "MARKUPLM_PRETRAINED_MODEL_ARCHIVE_LIST",
         "MarkupLMForQuestionAnswering",
         "MarkupLMForSequenceClassification",
         "MarkupLMForTokenClassification",
         "MarkupLMModel",
         "MarkupLMPreTrainedModel",
     ]


if TYPE_CHECKING:
     from .configuration_markuplm import MARKUPLM_PRETRAINED_CONFIG_ARCHIVE_MAP, MarkupLMConfig
     from .feature_extraction_markuplm import MarkupLMFeatureExtractor
     from .processing_markuplm import MarkupLMProcessor
     from .tokenization_markuplm import MarkupLMTokenizer

     try:
          if not is_tokenizers_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .tokenization_markuplm_fast import MarkupLMTokenizerFast

     try:
          if not is_torch_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_markuplm import (
              MARKUPLM_PRETRAINED_MODEL_ARCHIVE_LIST,
              MarkupLMForQuestionAnswering,
              MarkupLMForSequenceClassification,
              MarkupLMForTokenClassification,
              MarkupLMModel,
              MarkupLMPreTrainedModel,
          )


else:
     import sys

     _UpperCAmelCase						:    Dict      =  _LazyModule(__name__, globals()["__file__"], _import_structure)









from unittest.mock import patch

import datasets
from datasets import Dataset



def    UpperCAmelCase__	(  ):
  lowercase     :Any   						=    Dataset.from_dict({"col_1": [3, 2, 0, 1]} )

  with patch("tqdm.auto.tqdm" ) as mock_tqdm:
    datasets.disable_progress_bar()
    dset.map(lambda lowerCamelCase : {"col_2": x["col_1"] + 1} )
    mock_tqdm.assert_not_called()

    mock_tqdm.reset_mock()

    datasets.enable_progress_bar()
    dset.map(lambda lowerCamelCase : {"col_2": x["col_1"] + 1} )
    mock_tqdm.assert_called()



