





from unittest.mock import patch

import datasets
from datasets import Dataset







def       lowerCAmelCase__(	) ->    List[Any]:
  __snake_case   :				int        =       Dataset.from_dict({"col_1": [3, 2, 0, 1]}				)

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

    mock_tqdm.reset_mock()

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








from typing import TYPE_CHECKING

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


_UpperCamelCase       = {
    '''configuration_deta''': ['''DETA_PRETRAINED_CONFIG_ARCHIVE_MAP''', '''DetaConfig'''],
}

try:
     if not is_vision_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     _UpperCamelCase       = ['''DetaImageProcessor''']

try:
     if not is_torch_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     _UpperCamelCase       = [
         '''DETA_PRETRAINED_MODEL_ARCHIVE_LIST''',
         '''DetaForObjectDetection''',
         '''DetaModel''',
         '''DetaPreTrainedModel''',
     ]


if TYPE_CHECKING:
     from .configuration_deta import DETA_PRETRAINED_CONFIG_ARCHIVE_MAP, DetaConfig

     try:
          if not is_vision_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .image_processing_deta import DetaImageProcessor

     try:
          if not is_torch_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_deta import (
              DETA_PRETRAINED_MODEL_ARCHIVE_LIST,
              DetaForObjectDetection,
              DetaModel,
              DetaPreTrainedModel,
          )

else:
     import sys

     _UpperCamelCase       = _LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)

