

"""simple docstring"""



from abc import ABC, abstractmethod
from typing import Optional, Union

from .. import Dataset, DatasetDict, Features, IterableDataset, IterableDatasetDict, NamedSplit
from ..utils.typing import NestedDataStructureLike, PathLike



class 					a						(       __SCREAMING_SNAKE_CASE				):



    """simple docstring"""



    def __init__(     self: int    ,						UpperCamelCase: Tuple = None    ,						UpperCamelCase: Optional[Any] = None    ,						UpperCamelCase: Any = None    ,						UpperCamelCase: Union[str, Any] = None    ,						UpperCamelCase: List[Any] = False    ,						UpperCamelCase: List[Any] = False    ,						UpperCamelCase: Union[str, Any] = None    ,						**UpperCamelCase: Dict    ,						):






         """simple docstring"""


         A__										=  path_or_paths
         A__										=  split if split or isinstance(_a    ,						_a       ) else """train"""
         A__										=  features
         A__										=  cache_dir
         A__										=  keep_in_memory
         A__										=  streaming
         A__										=  num_proc
         A__										=  kwargs





    @abstractmethod
    def 				UpperCamelCase	(     self: Optional[Any]       ):






         """simple docstring"""


         pass



class 					a						(       __SCREAMING_SNAKE_CASE				):



    """simple docstring"""



    def __init__(     self: Union[str, Any]    ,						UpperCamelCase: Union[str, Any] = None    ,						UpperCamelCase: int = None    ,						UpperCamelCase: Optional[int] = False    ,						UpperCamelCase: int = False    ,						UpperCamelCase: List[str] = None    ,						**UpperCamelCase: int    ,						):






         """simple docstring"""


         A__										=  features
         A__										=  cache_dir
         A__										=  keep_in_memory
         A__										=  streaming
         A__										=  num_proc
         A__										=  kwargs





    @abstractmethod
    def 				UpperCamelCase	(     self: int       ):






         """simple docstring"""


         pass









"""simple docstring"""



SCREAMING_SNAKE_CASE_					:	str							  =						'0.21.0'

from .accelerator import Accelerator
from .big_modeling import (
    cpu_offload,
    cpu_offload_with_hook,
    disk_offload,
    dispatch_model,
    init_empty_weights,
    init_on_device,
    load_checkpoint_and_dispatch,
)
from .data_loader import skip_first_batches
from .launchers import debug_launcher, notebook_launcher
from .state import PartialState
from .utils import (
    DeepSpeedPlugin,
    DistributedDataParallelKwargs,
    DistributedType,
    FullyShardedDataParallelPlugin,
    GradScalerKwargs,
    InitProcessGroupKwargs,
    find_executable_batch_size,
    infer_auto_device_map,
    is_rich_available,
    load_checkpoint_in_model,
    synchronize_rng_states,
)


if is_rich_available():
 from .utils import rich






