
"""simple docstring"""






from unittest.mock import patch

import datasets
from datasets import Dataset
def    UpperCamelCase						(					)						->Union[str, Any]:

   """simple docstring"""



   a_						   =     Dataset.from_dict({"col_1": [3, 2, 0, 1]}     )

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

      mock_tqdm.reset_mock()

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






import unittest
from pathlib import Path

from transformers import is_vision_available, load_tool
from transformers.testing_utils import get_tests_dir

from .test_tools_common import ToolTesterMixin


if is_vision_available():
       from PIL import Image

class 			__A(    unittest.TestCase    ,      __UpperCAmelCase			):

     def 						_UpperCamelCase  (   self   ):





            """simple docstring"""

            _UpperCamelCase														=    load_tool('''image-question-answering'''   )
            self.tool.setup()
            _UpperCamelCase														=    load_tool('''image-question-answering''',							remote=A   )

     def 						_UpperCamelCase  (   self   ):





            """simple docstring"""

            _UpperCamelCase														=    Image.open(Path(get_tests_dir('''fixtures/tests_samples/COCO'''   )   ) / '''000000039769.png'''   )
            _UpperCamelCase														=    self.tool(A,							'''How many cats are sleeping on the couch?'''   )
            self.assertEqual(A,							'''2'''   )

     def 						_UpperCamelCase  (   self   ):





            """simple docstring"""

            _UpperCamelCase														=    Image.open(Path(get_tests_dir('''fixtures/tests_samples/COCO'''   )   ) / '''000000039769.png'''   )
            _UpperCamelCase														=    self.remote_tool(A,							'''How many cats are sleeping on the couch?'''   )
            self.assertEqual(A,							'''2'''   )

     def 						_UpperCamelCase  (   self   ):





            """simple docstring"""

            _UpperCamelCase														=    Image.open(Path(get_tests_dir('''fixtures/tests_samples/COCO'''   )   ) / '''000000039769.png'''   )
            _UpperCamelCase														=    self.tool(image=A,							question='''How many cats are sleeping on the couch?'''   )
            self.assertEqual(A,							'''2'''   )



     def 						_UpperCamelCase  (   self   ):





            """simple docstring"""

            _UpperCamelCase														=    Image.open(Path(get_tests_dir('''fixtures/tests_samples/COCO'''   )   ) / '''000000039769.png'''   )
            _UpperCamelCase														=    self.remote_tool(image=A,							question='''How many cats are sleeping on the couch?'''   )
            self.assertEqual(A,							'''2'''   )






