





'''simple docstring'''




import PIL.Image
import PIL.ImageOps
from packaging import version
from PIL import Image


if version.parse(version.parse(PIL.__version__).base_version) >= version.parse('''9.1.0'''):
  lowercase__    :      List[Any]						=    {
      '''linear''': PIL.Image.Resampling.BILINEAR,
      '''bilinear''': PIL.Image.Resampling.BILINEAR,
      '''bicubic''': PIL.Image.Resampling.BICUBIC,
      '''lanczos''': PIL.Image.Resampling.LANCZOS,
      '''nearest''': PIL.Image.Resampling.NEAREST,
  }
else:
  lowercase__    :      int						=    {
      '''linear''': PIL.Image.LINEAR,
      '''bilinear''': PIL.Image.BILINEAR,
      '''bicubic''': PIL.Image.BICUBIC,
      '''lanczos''': PIL.Image.LANCZOS,
      '''nearest''': PIL.Image.NEAREST,
  }






def        _lowerCAmelCase			(  __snake_case  : Union[str, Any]			)     ->     Union[str, Any]:
     __A						: List[Any]      =       (images / 2 + 0.5).clamp(0 ,							1			)
     __A						: int      =       images.cpu().permute(0 ,							2 ,							3 ,							1			).float().numpy()
     __A						: Any      =       numpy_to_pil(__snake_case			)
     return images







def        _lowerCAmelCase			(  __snake_case  : Dict			)     ->     Tuple:
     if images.ndim == 3:
          __A						: str      =       images[None, ...]
     __A						: Union[str, Any]      =       (images * 2_55).round().astype('uint8'			)
     if images.shape[-1] == 1:
          # special case for grayscale (single channel) images
          __A						: List[Any]      =       [Image.fromarray(image.squeeze() ,							mode='L'			) for image in images]
     else:
          __A						: Optional[int]      =       [Image.fromarray(__snake_case			) for image in images]

     return pil_images






'''simple docstring'''




import itertools
from dataclasses import dataclass
from typing import Optional

import pandas as pd
import pyarrow as pa

import datasets
from datasets.table import table_cast


@dataclass
class  SCREAMING_SNAKE_CASE    (datasets.BuilderConfig							):
     lowerCAmelCase					=       None




class  SCREAMING_SNAKE_CASE    (datasets.ArrowBasedBuilder							):
     lowerCAmelCase					=       PandasConfig


     def 							SCREAMING_SNAKE_CASE  (						self):

          '''simple docstring'''






          return datasets.DatasetInfo(features=self.config.features)


     def 							SCREAMING_SNAKE_CASE  (						self						,	_UpperCAmelCase):

          '''simple docstring'''






          if not self.config.data_files:
               raise ValueError(F'At least one data file must be specified, but got data_files={self.config.data_files}')
          __A						: Dict      =       dl_manager.download_and_extract(self.config.data_files)
          if isinstance(_UpperCAmelCase						,	(str, list, tuple)):
               __A						: Union[str, Any]      =       data_files
               if isinstance(_UpperCAmelCase						,	_UpperCAmelCase):
                    __A						: Optional[int]      =       [files]
               # Use `dl_manager.iter_files` to skip hidden files in an extracted archive
               __A						: Optional[Any]      =       [dl_manager.iter_files(_UpperCAmelCase) for file in files]
               return [datasets.SplitGenerator(name=datasets.Split.TRAIN						,	gen_kwargs={'files': files})]
          __A						: Tuple      =       []
          for split_name, files in data_files.items():
               if isinstance(_UpperCAmelCase						,	_UpperCAmelCase):
                    __A						: Dict      =       [files]
               # Use `dl_manager.iter_files` to skip hidden files in an extracted archive
               __A						: Optional[Any]      =       [dl_manager.iter_files(_UpperCAmelCase) for file in files]
               splits.append(datasets.SplitGenerator(name=_UpperCAmelCase						,	gen_kwargs={'files': files}))
          return splits


     def 							SCREAMING_SNAKE_CASE  (						self						,	_UpperCAmelCase):

          '''simple docstring'''






          if self.config.features is not None:
               # more expensive cast to support nested features with keys in a different order
               # allows str <-> int/float or str to Audio for example
               __A						: List[str]      =       table_cast(_UpperCAmelCase						,	self.config.features.arrow_schema)
          return pa_table


     def 							SCREAMING_SNAKE_CASE  (						self						,	_UpperCAmelCase):

          '''simple docstring'''






          for i, file in enumerate(itertools.chain.from_iterable(_UpperCAmelCase)):
               with open(_UpperCAmelCase						,	'rb') as f:
                    __A						: Optional[int]      =       pa.Table.from_pandas(pd.read_pickle(_UpperCAmelCase))
                    yield i, self._cast_table(_UpperCAmelCase)