






from ..utils import DummyObject, requires_backends


class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Optional[Any]:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Dict:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Optional[Any]:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Dict:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			List[str]:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			List[str]:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Optional[int]:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Union[str, Any]:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			str:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			int:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Any:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			int:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			List[Any]:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Any:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Dict:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			int:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Tuple:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Tuple:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			List[str]:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Dict:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Any:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Any:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			List[Any]:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Tuple:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Optional[Any]:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			int:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			List[str]:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Any:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			int:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			str:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Dict:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Union[str, Any]:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			List[str]:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Dict:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			str:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			List[Any]:
     requires_backends(cls		,		['flax']			)




class     _a							(	metaclass=UpperCAmelCase__     ):






  """simple docstring"""





  A_						       = ["""flax"""]




  def __init__(  self		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			Dict:
     requires_backends(self		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			List[str]:
     requires_backends(cls		,		['flax']			)




  @classmethod
  def  _UpperCAmelCase  (  cls		,		*_UpperCAmelCase		,		**_UpperCAmelCase			)      ->			List[str]:
     requires_backends(cls		,		['flax']			)









"""simple docstring"""



from typing import Optional

import pyspark

from .. import Features, NamedSplit
from ..download import DownloadMode
from ..packaged_modules.spark.spark import Spark
from .abc import AbstractDatasetReader







class    __a						( __snake_case							):







				def __init__( self      ,	UpperCAmelCase      ,	UpperCAmelCase = None      ,	UpperCAmelCase = None      ,	UpperCAmelCase = True      ,	UpperCAmelCase = None      ,	UpperCAmelCase = False      ,	UpperCAmelCase = None      ,	UpperCAmelCase = True      ,	UpperCAmelCase = "arrow"      ,	**UpperCAmelCase      ,	):

					'''simple docstring'''






					super().__init__(
					    split=UpperCAmelCase      ,	features=UpperCAmelCase      ,	cache_dir=UpperCAmelCase      ,	keep_in_memory=UpperCAmelCase      ,	streaming=UpperCAmelCase      ,	**UpperCAmelCase      ,	)
					lowerCAmelCase_      							=			load_from_cache_file
					lowerCAmelCase_      							=			file_format
					lowerCAmelCase_      							=			Spark(
					    df=UpperCAmelCase      ,	features=UpperCAmelCase      ,	cache_dir=UpperCAmelCase      ,	working_dir=UpperCAmelCase      ,	**UpperCAmelCase      ,	)







				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					if self.streaming:
						return self.builder.as_streaming_dataset(split=self.split   )
					lowerCAmelCase_      							=			None if self._load_from_cache_file else DownloadMode.FORCE_REDOWNLOAD
					self.builder.download_and_prepare(
					    download_mode=UpperCAmelCase      ,	file_format=self._file_format      ,	)
					return self.builder.as_dataset(split=self.split   )