
"""simple docstring"""
from typing import TypeVar

from .arrow_dataset import Dataset, _split_by_node_map_style_dataset
from .iterable_dataset import IterableDataset, _split_by_node_iterable_dataset


lowercase		:    str			=       TypeVar('DatasetType', Dataset, IterableDataset)


def      lowerCamelCase_     (						lowerCamelCase_						,					lowerCamelCase_						,					lowerCamelCase_			)  ->      DatasetType:


					'''simple docstring'''
					if isinstance(lowerCamelCase_						,					lowerCamelCase_			):
										return _split_by_node_map_style_dataset(lowerCamelCase_						,					rank=lowerCamelCase_						,					world_size=lowerCamelCase_			)
					else:
										return _split_by_node_iterable_dataset(lowerCamelCase_						,					rank=lowerCamelCase_						,					world_size=lowerCamelCase_			)


"""simple docstring"""
from transformers import ProcessorMixin




class  _snake_case	(    _lowercase							):
							SCREAMING_SNAKE_CASE		  =    '''AutoFeatureExtractor'''
							SCREAMING_SNAKE_CASE		  =    '''AutoTokenizer'''
