




import importlib
import json
import os
from collections import OrderedDict
from typing import Dict, Optional, Union

# Build the list of all image processors
from ...configuration_utils import PretrainedConfig
from ...dynamic_module_utils import get_class_from_dynamic_module, resolve_trust_remote_code
from ...image_processing_utils import ImageProcessingMixin
from ...utils import CONFIG_NAME, IMAGE_PROCESSOR_NAME, get_file_from_repo, logging
from .auto_factory import _LazyAutoMapping
from .configuration_auto import (
    CONFIG_MAPPING_NAMES,
    AutoConfig,
    model_type_to_module_name,
    replace_list_option_in_docstrings,
)


lowerCAmelCase_					    = logging.get_logger(__name__)

lowerCAmelCase_					    = OrderedDict(
    [
        ("""align""", """EfficientNetImageProcessor"""),
        ("""beit""", """BeitImageProcessor"""),
        ("""bit""", """BitImageProcessor"""),
        ("""blip""", """BlipImageProcessor"""),
        ("""blip-2""", """BlipImageProcessor"""),
        ("""bridgetower""", """BridgeTowerImageProcessor"""),
        ("""chinese_clip""", """ChineseCLIPImageProcessor"""),
        ("""clip""", """CLIPImageProcessor"""),
        ("""clipseg""", """ViTImageProcessor"""),
        ("""conditional_detr""", """ConditionalDetrImageProcessor"""),
        ("""convnext""", """ConvNextImageProcessor"""),
        ("""convnextv2""", """ConvNextImageProcessor"""),
        ("""cvt""", """ConvNextImageProcessor"""),
        ("""data2vec-vision""", """BeitImageProcessor"""),
        ("""deformable_detr""", """DeformableDetrImageProcessor"""),
        ("""deit""", """DeiTImageProcessor"""),
        ("""deta""", """DetaImageProcessor"""),
        ("""detr""", """DetrImageProcessor"""),
        ("""dinat""", """ViTImageProcessor"""),
        ("""donut-swin""", """DonutImageProcessor"""),
        ("""dpt""", """DPTImageProcessor"""),
        ("""efficientformer""", """EfficientFormerImageProcessor"""),
        ("""efficientnet""", """EfficientNetImageProcessor"""),
        ("""flava""", """FlavaImageProcessor"""),
        ("""focalnet""", """BitImageProcessor"""),
        ("""git""", """CLIPImageProcessor"""),
        ("""glpn""", """GLPNImageProcessor"""),
        ("""groupvit""", """CLIPImageProcessor"""),
        ("""imagegpt""", """ImageGPTImageProcessor"""),
        ("""instructblip""", """BlipImageProcessor"""),
        ("""layoutlmv2""", """LayoutLMv2ImageProcessor"""),
        ("""layoutlmv3""", """LayoutLMv3ImageProcessor"""),
        ("""levit""", """LevitImageProcessor"""),
        ("""mask2former""", """Mask2FormerImageProcessor"""),
        ("""maskformer""", """MaskFormerImageProcessor"""),
        ("""mgp-str""", """ViTImageProcessor"""),
        ("""mobilenet_v1""", """MobileNetV1ImageProcessor"""),
        ("""mobilenet_v2""", """MobileNetV2ImageProcessor"""),
        ("""mobilevit""", """MobileViTImageProcessor"""),
        ("""mobilevit""", """MobileViTImageProcessor"""),
        ("""mobilevitv2""", """MobileViTImageProcessor"""),
        ("""nat""", """ViTImageProcessor"""),
        ("""oneformer""", """OneFormerImageProcessor"""),
        ("""owlvit""", """OwlViTImageProcessor"""),
        ("""perceiver""", """PerceiverImageProcessor"""),
        ("""pix2struct""", """Pix2StructImageProcessor"""),
        ("""poolformer""", """PoolFormerImageProcessor"""),
        ("""regnet""", """ConvNextImageProcessor"""),
        ("""resnet""", """ConvNextImageProcessor"""),
        ("""sam""", """SamImageProcessor"""),
        ("""segformer""", """SegformerImageProcessor"""),
        ("""swiftformer""", """ViTImageProcessor"""),
        ("""swin""", """ViTImageProcessor"""),
        ("""swin2sr""", """Swin2SRImageProcessor"""),
        ("""swinv2""", """ViTImageProcessor"""),
        ("""table-transformer""", """DetrImageProcessor"""),
        ("""timesformer""", """VideoMAEImageProcessor"""),
        ("""tvlt""", """TvltImageProcessor"""),
        ("""upernet""", """SegformerImageProcessor"""),
        ("""van""", """ConvNextImageProcessor"""),
        ("""videomae""", """VideoMAEImageProcessor"""),
        ("""vilt""", """ViltImageProcessor"""),
        ("""vit""", """ViTImageProcessor"""),
        ("""vit_hybrid""", """ViTHybridImageProcessor"""),
        ("""vit_mae""", """ViTImageProcessor"""),
        ("""vit_msn""", """ViTImageProcessor"""),
        ("""xclip""", """CLIPImageProcessor"""),
        ("""yolos""", """YolosImageProcessor"""),
    ]
)

lowerCAmelCase_					    = _LazyAutoMapping(CONFIG_MAPPING_NAMES, IMAGE_PROCESSOR_MAPPING_NAMES)



def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Any:
  for module_name, extractors in IMAGE_PROCESSOR_MAPPING_NAMES.items():
    if class_name in extractors:
      lowerCAmelCase__				:  Tuple		       =     model_type_to_module_name(UpperCamelCase      )

      lowerCAmelCase__				:  int		       =     importlib.import_module(F""".{module_name}"""	,      '''transformers.models'''      )
      try:
        return getattr(UpperCamelCase	,      UpperCamelCase      )
      except AttributeError:
        continue

  for _, extractor in IMAGE_PROCESSOR_MAPPING._extra_content.items():
    if getattr(UpperCamelCase	,      '''__name__'''	,      UpperCamelCase      ) == class_name:
      return extractor

    # We did not fine the class, but maybe it's because a dep is missing. In that case, the class will be in the main
    # init and we return the proper dummy to get an appropriate error message.
  lowerCAmelCase__				:  Optional[int]		       =     importlib.import_module('''transformers'''      )
  if hasattr(UpperCamelCase	,      UpperCamelCase      ):
    return getattr(UpperCamelCase	,      UpperCamelCase      )

  return None



def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase = None	,      UpperCamelCase = False	,      UpperCamelCase = False	,      UpperCamelCase = None	,      UpperCamelCase = None	,      UpperCamelCase = None	,      UpperCamelCase = False	,      **UpperCamelCase	,      )     ->	Optional[int]:
  lowerCAmelCase__				:  Dict		       =     get_file_from_repo(
      UpperCamelCase	,      UpperCamelCase	,      cache_dir=UpperCamelCase	,      force_download=UpperCamelCase	,      resume_download=UpperCamelCase	,      proxies=UpperCamelCase	,      use_auth_token=UpperCamelCase	,      revision=UpperCamelCase	,      local_files_only=UpperCamelCase	,      )
  if resolved_config_file is None:
    logger.info(
        '''Could not locate the image processor configuration file, will try to use the model config instead.'''      )
    return {}

  with open(UpperCamelCase	,      encoding='''utf-8'''      ) as reader:
    return json.load(UpperCamelCase      )






class  _lowerCAmelCase     :

      def __init__(   self		):
        raise EnvironmentError(
            '''AutoImageProcessor is designed to be instantiated '''
            '''using the `AutoImageProcessor.from_pretrained(pretrained_model_name_or_path)` method.'''		)

      @classmethod
      @replace_list_option_in_docstrings(__UpperCAmelCase		)
      def 	__magic_name__(   cls      ,				__UpperCAmelCase      ,				**__UpperCAmelCase		):
        lowerCAmelCase__				:  Union[str, Any]		       =     kwargs.pop('''config'''      ,				__UpperCAmelCase		)
        lowerCAmelCase__				:  List[str]		       =     kwargs.pop('''trust_remote_code'''      ,				__UpperCAmelCase		)
        lowerCAmelCase__				:  List[str]		       =     True

        lowerCAmelCase__       ,			lowerCAmelCase__				:  Dict		       =     ImageProcessingMixin.get_image_processor_dict(__UpperCAmelCase      ,				**__UpperCAmelCase		)
        lowerCAmelCase__				:  int		       =     config_dict.get('''image_processor_type'''      ,				__UpperCAmelCase		)
        lowerCAmelCase__				:  str		       =     None
        if "AutoImageProcessor" in config_dict.get('''auto_map'''      ,				{}		):
          lowerCAmelCase__				:  Dict		       =     config_dict['''auto_map''']['''AutoImageProcessor''']

        # If we still don't have the image processor class, check if we're loading from a previous feature extractor config
        # and if so, infer the image processor class from there.
        if image_processor_class is None and image_processor_auto_map is None:
          lowerCAmelCase__				:  int		       =     config_dict.pop('''feature_extractor_type'''      ,				__UpperCAmelCase		)
          if feature_extractor_class is not None:
            logger.warning(
                '''Could not find image processor class in the image processor config or the model config. Loading'''
                ''' based on pattern matching with the model\'s feature extractor configuration.'''		)
            lowerCAmelCase__				:  Optional[int]		       =     feature_extractor_class.replace('''FeatureExtractor'''      ,				'''ImageProcessor'''		)
          if "AutoFeatureExtractor" in config_dict.get('''auto_map'''      ,				{}		):
            lowerCAmelCase__				:  Dict		       =     config_dict['''auto_map''']['''AutoFeatureExtractor''']
            lowerCAmelCase__				:  Optional[int]		       =     feature_extractor_auto_map.replace('''FeatureExtractor'''      ,				'''ImageProcessor'''		)
            logger.warning(
                '''Could not find image processor auto map in the image processor config or the model config.'''
                ''' Loading based on pattern matching with the model\'s feature extractor configuration.'''		)

        # If we don't find the image processor class in the image processor config, let's try the model config.
        if image_processor_class is None and image_processor_auto_map is None:
          if not isinstance(__UpperCAmelCase      ,				__UpperCAmelCase		):
            lowerCAmelCase__				:  Optional[int]		       =     AutoConfig.from_pretrained(__UpperCAmelCase      ,				**__UpperCAmelCase		)
          # It could be in `config.image_processor_type``
          lowerCAmelCase__				:  List[str]		       =     getattr(__UpperCAmelCase      ,				'''image_processor_type'''      ,				__UpperCAmelCase		)
          if hasattr(__UpperCAmelCase      ,				'''auto_map'''		) and "AutoImageProcessor" in config.auto_map:
            lowerCAmelCase__				:  Tuple		       =     config.auto_map['''AutoImageProcessor''']

        if image_processor_class is not None:
          lowerCAmelCase__				:  Union[str, Any]		       =     image_processor_class_from_name(__UpperCAmelCase		)

        lowerCAmelCase__				:  Union[str, Any]		       =     image_processor_auto_map is not None
        lowerCAmelCase__				:  Dict		       =     image_processor_class is not None or type(__UpperCAmelCase		) in IMAGE_PROCESSOR_MAPPING
        lowerCAmelCase__				:  Tuple		       =     resolve_trust_remote_code(
            __UpperCAmelCase      ,				__UpperCAmelCase      ,				__UpperCAmelCase      ,				__UpperCAmelCase		)

        if has_remote_code and trust_remote_code:
          lowerCAmelCase__				:  str		       =     get_class_from_dynamic_module(
              __UpperCAmelCase      ,				__UpperCAmelCase      ,				**__UpperCAmelCase		)
          lowerCAmelCase__				:  Any		       =     kwargs.pop('''code_revision'''      ,				__UpperCAmelCase		)
          if os.path.isdir(__UpperCAmelCase		):
            image_processor_class.register_for_auto_class()
          return image_processor_class.from_dict(__UpperCAmelCase      ,				**__UpperCAmelCase		)
        elif image_processor_class is not None:
          return image_processor_class.from_dict(__UpperCAmelCase      ,				**__UpperCAmelCase		)
        # Last try: we use the IMAGE_PROCESSOR_MAPPING.
        elif type(__UpperCAmelCase		) in IMAGE_PROCESSOR_MAPPING:
          lowerCAmelCase__				:  Dict		       =     IMAGE_PROCESSOR_MAPPING[type(__UpperCAmelCase		)]
          return image_processor_class.from_dict(__UpperCAmelCase      ,				**__UpperCAmelCase		)

        raise ValueError(
            f"""Unrecognized image processor in {pretrained_model_name_or_path}. Should have a """
            f"""`image_processor_type` key in its {IMAGE_PROCESSOR_NAME} of {CONFIG_NAME}, or one of the following """
            f"""`model_type` keys in its {CONFIG_NAME}: {", ".join(c for c in IMAGE_PROCESSOR_MAPPING_NAMES.keys()		)}"""		)







      @staticmethod
      def 	__magic_name__(   __UpperCAmelCase      ,				__UpperCAmelCase		):
        IMAGE_PROCESSOR_MAPPING.register(__UpperCAmelCase      ,				__UpperCAmelCase		)







import contextlib
import csv
import json
import os
import sqlitea
import tarfile
import textwrap
import zipfile

import pyarrow as pa
import pyarrow.parquet as pq
import pytest

import datasets
import datasets.config



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					)     ->	Optional[Any]:
  lowerCAmelCase__				:  List[str]		       =     10
  lowerCAmelCase__				:  Optional[Any]		       =     datasets.Features(
      {
          '''tokens''': datasets.Sequence(datasets.Value('''string'''      )      ),
          '''labels''': datasets.Sequence(datasets.ClassLabel(names=['''negative''', '''positive''']      )      ),
          '''answers''': datasets.Sequence(
              {
                  '''text''': datasets.Value('''string'''      ),
                  '''answer_start''': datasets.Value('''int32'''      ),
              }      ),
          '''id''': datasets.Value('''int64'''      ),
      }      )
  lowerCAmelCase__				:  List[Any]		       =     datasets.Dataset.from_dict(
      {
          '''tokens''': [['''foo'''] * 5] * n,
          '''labels''': [[1] * 5] * n,
          '''answers''': [{'''answer_start''': [97], '''text''': ['''1976''']}] * 10,
          '''id''': list(range(UpperCamelCase      )      ),
      }	,      features=UpperCamelCase	,      )
  return dataset



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase      )     ->	List[Any]:
  lowerCAmelCase__				:  str		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''file.arrow'''      )
  dataset.map(cache_file_name=UpperCamelCase      )
  return filename


# FILE_CONTENT + files


lowerCAmelCase_					    = """\
    Text data.
    Second line of data."""



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Optional[int]:
  lowerCAmelCase__				:  Union[str, Any]		       =     tmp_path_factory.mktemp('''data'''      ) / '''file.txt'''
  lowerCAmelCase__				:  int		       =     FILE_CONTENT
  with open(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase      )
  return filename



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Optional[Any]:
  import bza

  lowerCAmelCase__				:  List[Any]		       =     tmp_path_factory.mktemp('''data'''      ) / '''file.txt.bz2'''
  lowerCAmelCase__				:  Optional[Any]		       =     bytes(UpperCamelCase	,      '''utf-8'''      )
  with bza.open(UpperCamelCase	,      '''wb'''      ) as f:
    f.write(UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	List[Any]:
  import gzip

  lowerCAmelCase__				:  List[str]		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''file.txt.gz'''      )
  lowerCAmelCase__				:  Any		       =     bytes(UpperCamelCase	,      '''utf-8'''      )
  with gzip.open(UpperCamelCase	,      '''wb'''      ) as f:
    f.write(UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Tuple:
  if datasets.config.LZ4_AVAILABLE:
    import lza.frame

    lowerCAmelCase__				:  Union[str, Any]		       =     tmp_path_factory.mktemp('''data'''      ) / '''file.txt.lz4'''
    lowerCAmelCase__				:  Tuple		       =     bytes(UpperCamelCase	,      '''utf-8'''      )
    with lza.frame.open(UpperCamelCase	,      '''wb'''      ) as f:
      f.write(UpperCamelCase      )
    return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase      )     ->	Optional[Any]:
  if datasets.config.PY7ZR_AVAILABLE:
    import pyazr

    lowerCAmelCase__				:  int		       =     tmp_path_factory.mktemp('''data'''      ) / '''file.txt.7z'''
    with pyazr.SevenZipFile(UpperCamelCase	,      '''w'''      ) as archive:
      archive.write(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
    return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase      )     ->	Union[str, Any]:
  import tarfile

  lowerCAmelCase__				:  List[str]		       =     tmp_path_factory.mktemp('''data'''      ) / '''file.txt.tar'''
  with tarfile.TarFile(UpperCamelCase	,      '''w'''      ) as f:
    f.add(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Optional[int]:
  import lzma

  lowerCAmelCase__				:  Union[str, Any]		       =     tmp_path_factory.mktemp('''data'''      ) / '''file.txt.xz'''
  lowerCAmelCase__				:  str		       =     bytes(UpperCamelCase	,      '''utf-8'''      )
  with lzma.open(UpperCamelCase	,      '''wb'''      ) as f:
    f.write(UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase      )     ->	Any:
  import zipfile

  lowerCAmelCase__				:  Optional[int]		       =     tmp_path_factory.mktemp('''data'''      ) / '''file.txt.zip'''
  with zipfile.ZipFile(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	List[str]:
  if datasets.config.ZSTANDARD_AVAILABLE:
    import zstandard as zstd

    lowerCAmelCase__				:  Tuple		       =     tmp_path_factory.mktemp('''data'''      ) / '''file.txt.zst'''
    lowerCAmelCase__				:  int		       =     bytes(UpperCamelCase	,      '''utf-8'''      )
    with zstd.open(UpperCamelCase	,      '''wb'''      ) as f:
      f.write(UpperCamelCase      )
    return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Dict:
  lowerCAmelCase__				:  Optional[int]		       =     tmp_path_factory.mktemp('''data'''      ) / '''file.xml'''
  lowerCAmelCase__				:  Tuple		       =     textwrap.dedent(
      '''\
    <?xml version="1.0" encoding="UTF-8" ?>
    <tmx version="1.4">
      <header segtype="sentence" srclang="ca" />
      <body>
        <tu>
          <tuv xml:lang="ca"><seg>Contingut 1</seg></tuv>
          <tuv xml:lang="en"><seg>Content 1</seg></tuv>
        </tu>
        <tu>
          <tuv xml:lang="ca"><seg>Contingut 2</seg></tuv>
          <tuv xml:lang="en"><seg>Content 2</seg></tuv>
        </tu>
        <tu>
          <tuv xml:lang="ca"><seg>Contingut 3</seg></tuv>
          <tuv xml:lang="en"><seg>Content 3</seg></tuv>
        </tu>
        <tu>
          <tuv xml:lang="ca"><seg>Contingut 4</seg></tuv>
          <tuv xml:lang="en"><seg>Content 4</seg></tuv>
        </tu>
        <tu>
          <tuv xml:lang="ca"><seg>Contingut 5</seg></tuv>
          <tuv xml:lang="en"><seg>Content 5</seg></tuv>
        </tu>
      </body>
    </tmx>'''      )
  with open(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase      )
  return filename


lowerCAmelCase_					    = [
    {"""col_1""": """0""", """col_2""": 0, """col_3""": 0.0},
    {"""col_1""": """1""", """col_2""": 1, """col_3""": 1.0},
    {"""col_1""": """2""", """col_2""": 2, """col_3""": 2.0},
    {"""col_1""": """3""", """col_2""": 3, """col_3""": 3.0},
]
lowerCAmelCase_					    = [
    {"""col_1""": """4""", """col_2""": 4, """col_3""": 4.0},
    {"""col_1""": """5""", """col_2""": 5, """col_3""": 5.0},
]
lowerCAmelCase_					    = {
    """col_1""": ["""0""", """1""", """2""", """3"""],
    """col_2""": [0, 1, 2, 3],
    """col_3""": [0.0, 1.0, 2.0, 3.0],
}

lowerCAmelCase_					    = [
    {"""col_3""": 0.0, """col_1""": """0""", """col_2""": 0},
    {"""col_3""": 1.0, """col_1""": """1""", """col_2""": 1},
]

lowerCAmelCase_					    = [
    {"""col_1""": """s0""", """col_2""": 0, """col_3""": 0.0},
    {"""col_1""": """s1""", """col_2""": 1, """col_3""": 1.0},
    {"""col_1""": """s2""", """col_2""": 2, """col_3""": 2.0},
    {"""col_1""": """s3""", """col_2""": 3, """col_3""": 3.0},
]



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					)     ->	Tuple:
  return DATA_DICT_OF_LISTS



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Any:
  lowerCAmelCase__				:  str		       =     datasets.Dataset.from_dict(UpperCamelCase      )
  lowerCAmelCase__				:  List[str]		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset.arrow'''      )
  dataset.map(cache_file_name=UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Any:
  lowerCAmelCase__				:  List[str]		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset.sqlite'''      )
  with contextlib.closing(sqlitea.connect(UpperCamelCase      )      ) as con:
    lowerCAmelCase__				:  int		       =     con.cursor()
    cur.execute('''CREATE TABLE dataset(col_1 text, col_2 int, col_3 real)'''      )
    for item in DATA:
      cur.execute('''INSERT INTO dataset(col_1, col_2, col_3) VALUES (?, ?, ?)'''	,      tuple(item.values()      )      )
    con.commit()
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	int:
  lowerCAmelCase__				:  List[str]		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset.csv'''      )
  with open(UpperCamelCase	,      '''w'''	,      newline=''''''      ) as f:
    lowerCAmelCase__				:  List[str]		       =     csv.DictWriter(UpperCamelCase	,      fieldnames=['''col_1''', '''col_2''', '''col_3''']      )
    writer.writeheader()
    for item in DATA:
      writer.writerow(UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	List[Any]:
  lowerCAmelCase__				:  Dict		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset2.csv'''      )
  with open(UpperCamelCase	,      '''w'''	,      newline=''''''      ) as f:
    lowerCAmelCase__				:  Tuple		       =     csv.DictWriter(UpperCamelCase	,      fieldnames=['''col_1''', '''col_2''', '''col_3''']      )
    writer.writeheader()
    for item in DATA:
      writer.writerow(UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase      )     ->	List[str]:
  import bza

  lowerCAmelCase__				:  Optional[int]		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset.csv.bz2'''
  with open(UpperCamelCase	,      '''rb'''      ) as f:
    lowerCAmelCase__				:  List[str]		       =     f.read()
  # data = bytes(FILE_CONTENT, "utf-8")
  with bza.open(UpperCamelCase	,      '''wb'''      ) as f:
    f.write(UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase      )     ->	Tuple:
  lowerCAmelCase__				:  Optional[int]		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset.csv.zip'''
  with zipfile.ZipFile(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
    f.write(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase      )     ->	Tuple:
  lowerCAmelCase__				:  Union[str, Any]		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset.csv.zip'''
  with zipfile.ZipFile(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase	,      arcname=os.path.basename(csv_path.replace('''.csv'''	,      '''.CSV'''      )      )      )
    f.write(UpperCamelCase	,      arcname=os.path.basename(csva_path.replace('''.csv'''	,      '''.CSV'''      )      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase      )     ->	str:
  lowerCAmelCase__				:  Union[str, Any]		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset_with_dir.csv.zip'''
  with zipfile.ZipFile(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase	,      arcname=os.path.join('''main_dir'''	,      os.path.basename(UpperCamelCase      )      )      )
    f.write(UpperCamelCase	,      arcname=os.path.join('''main_dir'''	,      os.path.basename(UpperCamelCase      )      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Dict:
  lowerCAmelCase__				:  str		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset.parquet'''      )
  lowerCAmelCase__				:  List[Any]		       =     pa.schema(
      {
          '''col_1''': pa.string(),
          '''col_2''': pa.intaa(),
          '''col_3''': pa.floataa(),
      }      )
  with open(UpperCamelCase	,      '''wb'''      ) as f:
    lowerCAmelCase__				:  str		       =     pq.ParquetWriter(UpperCamelCase	,      schema=UpperCamelCase      )
    lowerCAmelCase__				:  List[Any]		       =     pa.Table.from_pydict({k: [DATA[i][k] for i in range(len(UpperCamelCase      )      )] for k in DATA[0]}	,      schema=UpperCamelCase      )
    writer.write_table(UpperCamelCase      )
    writer.close()
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Dict:
  lowerCAmelCase__				:  List[Any]		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset.json'''      )
  lowerCAmelCase__				:  List[str]		       =     {'''data''': DATA}
  with open(UpperCamelCase	,      '''w'''      ) as f:
    json.dump(UpperCamelCase	,      UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Tuple:
  lowerCAmelCase__				:  Tuple		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset.json'''      )
  lowerCAmelCase__				:  Optional[Any]		       =     {'''data''': DATA_DICT_OF_LISTS}
  with open(UpperCamelCase	,      '''w'''      ) as f:
    json.dump(UpperCamelCase	,      UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	int:
  lowerCAmelCase__				:  Union[str, Any]		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset.jsonl'''      )
  with open(UpperCamelCase	,      '''w'''      ) as f:
    for item in DATA:
      f.write(json.dumps(UpperCamelCase      ) + '''\n'''      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	str:
  lowerCAmelCase__				:  Optional[int]		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset2.jsonl'''      )
  with open(UpperCamelCase	,      '''w'''      ) as f:
    for item in DATA:
      f.write(json.dumps(UpperCamelCase      ) + '''\n'''      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Optional[Any]:
  lowerCAmelCase__				:  Dict		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset_312.jsonl'''      )
  with open(UpperCamelCase	,      '''w'''      ) as f:
    for item in DATA_312:
      f.write(json.dumps(UpperCamelCase      ) + '''\n'''      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	List[Any]:
  lowerCAmelCase__				:  Optional[int]		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset-str.jsonl'''      )
  with open(UpperCamelCase	,      '''w'''      ) as f:
    for item in DATA_STR:
      f.write(json.dumps(UpperCamelCase      ) + '''\n'''      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase      )     ->	int:
  import gzip

  lowerCAmelCase__				:  Dict		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset.txt.gz'''      )
  with open(UpperCamelCase	,      '''rb'''      ) as orig_file:
    with gzip.open(UpperCamelCase	,      '''wb'''      ) as zipped_file:
      zipped_file.writelines(UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase      )     ->	List[Any]:
  import gzip

  lowerCAmelCase__				:  int		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset.jsonl.gz'''      )
  with open(UpperCamelCase	,      '''rb'''      ) as orig_file:
    with gzip.open(UpperCamelCase	,      '''wb'''      ) as zipped_file:
      zipped_file.writelines(UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase      )     ->	Union[str, Any]:
  lowerCAmelCase__				:  str		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset.jsonl.zip'''
  with zipfile.ZipFile(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
    f.write(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase      )     ->	Optional[int]:
  lowerCAmelCase__				:  List[Any]		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset_nested.jsonl.zip'''
  with zipfile.ZipFile(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase	,      arcname=os.path.join('''nested'''	,      os.path.basename(UpperCamelCase      )      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase      )     ->	Optional[Any]:
  lowerCAmelCase__				:  Tuple		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset_with_dir.jsonl.zip'''
  with zipfile.ZipFile(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase	,      arcname=os.path.join('''main_dir'''	,      os.path.basename(UpperCamelCase      )      )      )
    f.write(UpperCamelCase	,      arcname=os.path.join('''main_dir'''	,      os.path.basename(UpperCamelCase      )      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase      )     ->	int:
  lowerCAmelCase__				:  Optional[Any]		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset.jsonl.tar'''
  with tarfile.TarFile(UpperCamelCase	,      '''w'''      ) as f:
    f.add(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
    f.add(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase      )     ->	Optional[int]:
  lowerCAmelCase__				:  Union[str, Any]		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset_nested.jsonl.tar'''
  with tarfile.TarFile(UpperCamelCase	,      '''w'''      ) as f:
    f.add(UpperCamelCase	,      arcname=os.path.join('''nested'''	,      os.path.basename(UpperCamelCase      )      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Optional[Any]:
  lowerCAmelCase__				:  Dict		       =     ['''0''', '''1''', '''2''', '''3''']
  lowerCAmelCase__				:  Any		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset.txt'''      )
  with open(UpperCamelCase	,      '''w'''      ) as f:
    for item in data:
      f.write(item + '''\n'''      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Any:
  lowerCAmelCase__				:  int		       =     ['''0''', '''1''', '''2''', '''3''']
  lowerCAmelCase__				:  int		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset2.txt'''      )
  with open(UpperCamelCase	,      '''w'''      ) as f:
    for item in data:
      f.write(item + '''\n'''      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Optional[int]:
  lowerCAmelCase__				:  Optional[Any]		       =     ['''0''', '''1''', '''2''', '''3''']
  lowerCAmelCase__				:  str		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset.abc'''
  with open(UpperCamelCase	,      '''w'''      ) as f:
    for item in data:
      f.write(item + '''\n'''      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase      )     ->	Any:
  lowerCAmelCase__				:  Optional[int]		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset.text.zip'''
  with zipfile.ZipFile(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
    f.write(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase      )     ->	Union[str, Any]:
  lowerCAmelCase__				:  Any		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset_with_dir.text.zip'''
  with zipfile.ZipFile(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase	,      arcname=os.path.join('''main_dir'''	,      os.path.basename(UpperCamelCase      )      )      )
    f.write(UpperCamelCase	,      arcname=os.path.join('''main_dir'''	,      os.path.basename(UpperCamelCase      )      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase	,      UpperCamelCase      )     ->	int:
  lowerCAmelCase__				:  str		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset.ext.zip'''
  with zipfile.ZipFile(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase	,      arcname=os.path.basename('''unsupported.ext'''      )      )
    f.write(UpperCamelCase	,      arcname=os.path.basename('''unsupported_2.ext'''      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Dict:
  lowerCAmelCase__				:  List[Any]		       =     '''\n'''.join(['''First''', '''Second\u2029with Unicode new line''', '''Third''']      )
  lowerCAmelCase__				:  Union[str, Any]		       =     str(tmp_path_factory.mktemp('''data'''      ) / '''dataset_with_unicode_new_lines.txt'''      )
  with open(UpperCamelCase	,      '''w'''	,      encoding='''utf-8'''      ) as f:
    f.write(UpperCamelCase      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					)     ->	Optional[int]:
  return os.path.join('''tests'''	,      '''features'''	,      '''data'''	,      '''test_image_rgb.jpg'''      )



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					)     ->	Optional[Any]:
  return os.path.join('''tests'''	,      '''features'''	,      '''data'''	,      '''test_audio_44100.wav'''      )



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase	,      UpperCamelCase      )     ->	Union[str, Any]:
  lowerCAmelCase__				:  Tuple		       =     tmp_path_factory.mktemp('''data'''      ) / '''dataset.img.zip'''
  with zipfile.ZipFile(UpperCamelCase	,      '''w'''      ) as f:
    f.write(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      )      )
    f.write(UpperCamelCase	,      arcname=os.path.basename(UpperCamelCase      ).replace('''.jpg'''	,      '''2.jpg'''      )      )
  return path



@pytest.fixture(scope='''session'''      )
def      __lowerCAmelCase			(					UpperCamelCase      )     ->	Dict:
  lowerCAmelCase__				:  Optional[Any]		       =     tmp_path_factory.mktemp('''data_dir'''      )

  (data_dir / "subdir").mkdir()
  with open(data_dir / '''subdir''' / '''train.txt'''	,      '''w'''      ) as f:
    f.write('''foo\n''' * 10      )
  with open(data_dir / '''subdir''' / '''test.txt'''	,      '''w'''      ) as f:
    f.write('''bar\n''' * 10      )
  # hidden file
  with open(data_dir / '''subdir''' / '''.test.txt'''	,      '''w'''      ) as f:
    f.write('''bar\n''' * 10      )

  # hidden directory
  (data_dir / ".subdir").mkdir()
  with open(data_dir / '''.subdir''' / '''train.txt'''	,      '''w'''      ) as f:
    f.write('''foo\n''' * 10      )
  with open(data_dir / '''.subdir''' / '''test.txt'''	,      '''w'''      ) as f:
    f.write('''bar\n''' * 10      )

  return data_dir

