




from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_flax_available,
    is_tf_available,
    is_torch_available,
)


_UpperCAmelCase						:    Union[str, Any]      =  {
    "configuration_gptsan_japanese": ["GPTSAN_JAPANESE_PRETRAINED_CONFIG_ARCHIVE_MAP", "GPTSanJapaneseConfig"],
    "tokenization_gptsan_japanese": ["GPTSanJapaneseTokenizer"],
}

try:
     if not is_torch_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     _UpperCAmelCase						:    List[Any]      =  [
         "GPTSAN_JAPANESE_PRETRAINED_MODEL_ARCHIVE_LIST",
         "GPTSanJapaneseForConditionalGeneration",
         "GPTSanJapaneseModel",
         "GPTSanJapanesePreTrainedModel",
     ]
     _UpperCAmelCase						:    str      =  [
         "GPTSanJapaneseTokenizer",
     ]


if TYPE_CHECKING:
     from .configuration_gptsan_japanese import GPTSAN_JAPANESE_PRETRAINED_CONFIG_ARCHIVE_MAP, GPTSanJapaneseConfig
     from .tokenization_gptsan_japanese import GPTSanJapaneseTokenizer

     try:
          if not is_torch_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_gptsan_japanese import (
              GPTSAN_JAPANESE_PRETRAINED_MODEL_ARCHIVE_LIST,
              GPTSanJapaneseForConditionalGeneration,
              GPTSanJapaneseModel,
              GPTSanJapanesePreTrainedModel,
          )
          from .tokenization_gptsan_japanese import GPTSanJapaneseTokenizer


else:
     import sys

     _UpperCAmelCase						:    List[Any]      =  _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)









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()



