






'''simple docstring'''







from typing import TYPE_CHECKING

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


__lowerCAmelCase        =							{
    "configuration_clip": [
        "CLIP_PRETRAINED_CONFIG_ARCHIVE_MAP",
        "CLIPConfig",
        "CLIPOnnxConfig",
        "CLIPTextConfig",
        "CLIPVisionConfig",
    ],
    "processing_clip": ["CLIPProcessor"],
    "tokenization_clip": ["CLIPTokenizer"],
}

try:
  if not is_tokenizers_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __lowerCAmelCase        =							["CLIPTokenizerFast"]

try:
  if not is_vision_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __lowerCAmelCase        =							["CLIPFeatureExtractor"]
  __lowerCAmelCase        =							["CLIPImageProcessor"]

try:
  if not is_torch_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __lowerCAmelCase        =							[
      "CLIP_PRETRAINED_MODEL_ARCHIVE_LIST",
      "CLIPModel",
      "CLIPPreTrainedModel",
      "CLIPTextModel",
      "CLIPTextModelWithProjection",
      "CLIPVisionModel",
      "CLIPVisionModelWithProjection",
  ]

try:
  if not is_tf_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __lowerCAmelCase        =							[
      "TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST",
      "TFCLIPModel",
      "TFCLIPPreTrainedModel",
      "TFCLIPTextModel",
      "TFCLIPVisionModel",
  ]

try:
  if not is_flax_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __lowerCAmelCase        =							[
      "FlaxCLIPModel",
      "FlaxCLIPPreTrainedModel",
      "FlaxCLIPTextModel",
      "FlaxCLIPTextPreTrainedModel",
      "FlaxCLIPVisionModel",
      "FlaxCLIPVisionPreTrainedModel",
  ]


if TYPE_CHECKING:
  from .configuration_clip import (
      CLIP_PRETRAINED_CONFIG_ARCHIVE_MAP,
      CLIPConfig,
      CLIPOnnxConfig,
      CLIPTextConfig,
      CLIPVisionConfig,
  )
  from .processing_clip import CLIPProcessor
  from .tokenization_clip import CLIPTokenizer

  try:
    if not is_tokenizers_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .tokenization_clip_fast import CLIPTokenizerFast

  try:
    if not is_vision_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .feature_extraction_clip import CLIPFeatureExtractor
    from .image_processing_clip import CLIPImageProcessor

  try:
    if not is_torch_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_clip import (
        CLIP_PRETRAINED_MODEL_ARCHIVE_LIST,
        CLIPModel,
        CLIPPreTrainedModel,
        CLIPTextModel,
        CLIPTextModelWithProjection,
        CLIPVisionModel,
        CLIPVisionModelWithProjection,
    )

  try:
    if not is_tf_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_tf_clip import (
        TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST,
        TFCLIPModel,
        TFCLIPPreTrainedModel,
        TFCLIPTextModel,
        TFCLIPVisionModel,
    )

  try:
    if not is_flax_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_flax_clip import (
        FlaxCLIPModel,
        FlaxCLIPPreTrainedModel,
        FlaxCLIPTextModel,
        FlaxCLIPTextPreTrainedModel,
        FlaxCLIPVisionModel,
        FlaxCLIPVisionPreTrainedModel,
    )


else:
  import sys

  __lowerCAmelCase        =							_LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)












'''simple docstring'''







import multiprocessing
import os
from typing import BinaryIO, Optional, Union

import fsspec

from .. import Dataset, Features, NamedSplit, config
from ..formatting import query_table
from ..packaged_modules.json.json import Json
from ..utils import logging
from ..utils.typing import NestedDataStructureLike, PathLike
from .abc import AbstractDatasetReader




class   __SCREAMING_SNAKE_CASE       (__A       ):


  """simple docstring"""


  def __init__(			self   ,	UpperCamelCase__   ,	UpperCamelCase__ = None   ,	UpperCamelCase__ = None   ,	UpperCamelCase__ = None   ,	UpperCamelCase__ = False   ,	UpperCamelCase__ = False   ,	UpperCamelCase__ = None   ,	UpperCamelCase__ = None   ,	**UpperCamelCase__   ,	):






         """simple docstring"""


         super().__init__(
             UpperCamelCase__   ,	split=UpperCamelCase__   ,	features=UpperCamelCase__   ,	cache_dir=UpperCamelCase__   ,	keep_in_memory=UpperCamelCase__   ,	streaming=UpperCamelCase__   ,	num_proc=UpperCamelCase__   ,	**UpperCamelCase__   ,	)
         a_						     =			field
         a_						     =			path_or_paths if isinstance(UpperCamelCase__   ,	UpperCamelCase__	) else {self.split: path_or_paths}
         a_						     =			Json(
             cache_dir=UpperCamelCase__   ,	data_files=UpperCamelCase__   ,	features=UpperCamelCase__   ,	field=UpperCamelCase__   ,	**UpperCamelCase__   ,	)







  def    _a       (			self	):






         """simple docstring"""


         if self.streaming:
                a_						     =			self.builder.as_streaming_dataset(split=self.split	)
         # Build regular (map-style) dataset
         else:
                a_						     =			None
                a_						     =			None
                a_						     =			None
                a_						     =			None

                self.builder.download_and_prepare(
                    download_config=UpperCamelCase__   ,	download_mode=UpperCamelCase__   ,	verification_mode=UpperCamelCase__   ,	base_path=UpperCamelCase__   ,	num_proc=self.num_proc   ,	)
                a_						     =			self.builder.as_dataset(
                    split=self.split   ,	verification_mode=UpperCamelCase__   ,	in_memory=self.keep_in_memory	)
         return dataset





class   __SCREAMING_SNAKE_CASE       :


  """simple docstring"""


  def __init__(			self   ,	UpperCamelCase__   ,	UpperCamelCase__   ,	UpperCamelCase__ = None   ,	UpperCamelCase__ = None   ,	**UpperCamelCase__   ,	):






         """simple docstring"""


         if num_proc is not None and num_proc <= 0:
                raise ValueError(f'num_proc {num_proc} must be an integer > 0.'	)

         a_						     =			dataset
         a_						     =			path_or_buf
         a_						     =			batch_size if batch_size else config.DEFAULT_MAX_BATCH_SIZE
         a_						     =			num_proc
         a_						     =			'utf-8'
         a_						     =			to_json_kwargs







  def    _a       (			self	):






         """simple docstring"""


         a_						     =			self.to_json_kwargs.pop('path_or_buf'   ,	UpperCamelCase__	)
         a_						     =			self.to_json_kwargs.pop('orient'   ,	'records'	)
         a_						     =			self.to_json_kwargs.pop('lines'   ,	True if orient == 'records' else False	)
         a_						     =			self.to_json_kwargs.pop('index'   ,	False if orient in ['split', 'table'] else True	)
         a_						     =			self.to_json_kwargs.pop('compression'   ,	UpperCamelCase__	)

         if compression not in [None, "infer", "gzip", "bz2", "xz"]:
                raise NotImplementedError(f'`datasets` currently does not support {compression} compression'	)

         if isinstance(self.path_or_buf   ,	(str, bytes, os.PathLike)	):
                with fsspec.open(self.path_or_buf   ,	'wb'   ,	compression=UpperCamelCase__	) as buffer:
                       a_						     =			self._write(file_obj=UpperCamelCase__   ,	orient=UpperCamelCase__   ,	lines=UpperCamelCase__   ,	index=UpperCamelCase__   ,	**self.to_json_kwargs	)
         else:
                if compression:
                       raise NotImplementedError(
                           f'The compression parameter is not supported when writing to a buffer, but compression={compression}'
                           ' was passed. Please provide a local path instead.'	)
                a_						     =			self._write(
                    file_obj=self.path_or_buf   ,	orient=UpperCamelCase__   ,	lines=UpperCamelCase__   ,	index=UpperCamelCase__   ,	**self.to_json_kwargs	)
         return written







  def    _a       (			self   ,	UpperCamelCase__	):






         """simple docstring"""


         a_     ,	a_     ,	a_     ,	a_     ,	a_						     =			args

         a_						     =			query_table(
             table=self.dataset.data   ,	key=slice(UpperCamelCase__   ,	offset + self.batch_size	)   ,	indices=self.dataset._indices   ,	)
         a_						     =			batch.to_pandas().to_json(
             path_or_buf=UpperCamelCase__   ,	orient=UpperCamelCase__   ,	lines=UpperCamelCase__   ,	index=UpperCamelCase__   ,	**UpperCamelCase__	)
         if not json_str.endswith('\n'	):
                json_str += "\n"
         return json_str.encode(self.encoding	)







  def    _a       (			self   ,	UpperCamelCase__   ,	UpperCamelCase__   ,	UpperCamelCase__   ,	UpperCamelCase__   ,	**UpperCamelCase__   ,	):






         """simple docstring"""


         a_						     =			0

         if self.num_proc is None or self.num_proc == 1:
                for offset in logging.tqdm(
                    range(0   ,	len(self.dataset	)   ,	self.batch_size	)   ,	unit='ba'   ,	disable=not logging.is_progress_bar_enabled()   ,	desc='Creating json from Arrow format'   ,	):
                       a_						     =			self._batch_json((offset, orient, lines, index, to_json_kwargs)	)
                       written += file_obj.write(UpperCamelCase__	)
         else:
                a_     ,	a_						     =			len(self.dataset	), self.batch_size
                with multiprocessing.Pool(self.num_proc	) as pool:
                       for json_str in logging.tqdm(
                           pool.imap(
                               self._batch_json   ,	[(offset, orient, lines, index, to_json_kwargs) for offset in range(0   ,	UpperCamelCase__   ,	UpperCamelCase__	)]   ,	)   ,	total=(num_rows // batch_size) + 1 if num_rows % batch_size else num_rows // batch_size   ,	unit='ba'   ,	disable=not logging.is_progress_bar_enabled()   ,	desc='Creating json from Arrow format'   ,	):
                              written += file_obj.write(UpperCamelCase__	)

         return written




