



from typing import List, Optional, TypeVar

from .arrow_dataset import Dataset, _concatenate_map_style_datasets, _interleave_map_style_datasets
from .dataset_dict import DatasetDict, IterableDatasetDict
from .info import DatasetInfo
from .iterable_dataset import IterableDataset, _concatenate_iterable_datasets, _interleave_iterable_datasets
from .splits import NamedSplit
from .utils import logging
from .utils.py_utils import Literal


snake_case    :						Dict  =						logging.get_logger(__name__)


snake_case    :						Optional[Any]  =						TypeVar('DatasetType', Dataset, IterableDataset)
def 	snake_case__       (					__lowercase							,     __lowercase = None							,     __lowercase = None							,     __lowercase = None							,     __lowercase = None							,     __lowercase = "first_exhausted"							,     )		->		DatasetType:







						"""simple docstring"""

						from .arrow_dataset import Dataset
						from .iterable_dataset import IterableDataset

						if not datasets:
												raise ValueError("Unable to interleave an empty list of datasets."       )
						for i, dataset in enumerate(__lowercase       ):
												if not isinstance(__lowercase							,     (Dataset, IterableDataset)       ):
																		if isinstance(__lowercase							,     (DatasetDict, IterableDatasetDict)       ):
																								if not dataset:
																														raise ValueError(
																														    F'Expected a list of Dataset objects or a list of IterableDataset objects, but element at position {i} '
																														    "is an empty dataset dictionary."       )
																								raise ValueError(
																								    F'Dataset at position {i} has at least one split: {list(__lowercase       )}\n'
																								    F'Please pick one to interleave with the other datasets, for example: dataset[\'{next(iter(__lowercase       )       )}\']'       )
																		raise ValueError(
																		    F'Expected a list of Dataset objects or a list of IterableDataset objects, but element at position {i} is a {type(__lowercase       ).__name__}.'       )
												if i == 0:
																		A__	:							Dict						    = (
																		    (Dataset, IterableDataset) if isinstance(__lowercase							,     __lowercase       ) else (IterableDataset, Dataset)
																		)
												elif not isinstance(__lowercase							,     __lowercase       ):
																		raise ValueError(
																		    F'Unable to interleave a {dataset_type.__name__} (at position 0) with a {other_type.__name__} (at position {i}). Expected a list of Dataset objects or a list of IterableDataset objects.'       )
						if stopping_strategy not in ["first_exhausted", "all_exhausted"]:
												raise ValueError(F'{stopping_strategy} is not supported. Please enter a valid stopping_strategy.'       )
						if dataset_type is Dataset:
												return _interleave_map_style_datasets(
												    __lowercase							,     __lowercase							,     __lowercase							,     info=__lowercase							,     split=__lowercase							,     stopping_strategy=__lowercase       )
						else:
												return _interleave_iterable_datasets(
												    __lowercase							,     __lowercase							,     __lowercase							,     info=__lowercase							,     split=__lowercase							,     stopping_strategy=__lowercase       )




def 	snake_case__       (					__lowercase							,     __lowercase = None							,     __lowercase = None							,     __lowercase = 0							,     )		->		DatasetType:







						"""simple docstring"""

						if not dsets:
												raise ValueError("Unable to concatenate an empty list of datasets."       )
						for i, dataset in enumerate(__lowercase       ):
												if not isinstance(__lowercase							,     (Dataset, IterableDataset)       ):
																		if isinstance(__lowercase							,     (DatasetDict, IterableDatasetDict)       ):
																								if not dataset:
																														raise ValueError(
																														    F'Expected a list of Dataset objects or a list of IterableDataset objects, but element at position {i} '
																														    "is an empty dataset dictionary."       )
																								raise ValueError(
																								    F'Dataset at position {i} has at least one split: {list(__lowercase       )}\n'
																								    F'Please pick one to interleave with the other datasets, for example: dataset[\'{next(iter(__lowercase       )       )}\']'       )
																		raise ValueError(
																		    F'Expected a list of Dataset objects or a list of IterableDataset objects, but element at position {i} is a {type(__lowercase       ).__name__}.'       )
												if i == 0:
																		A__	:							int						    = (
																		    (Dataset, IterableDataset) if isinstance(__lowercase							,     __lowercase       ) else (IterableDataset, Dataset)
																		)
												elif not isinstance(__lowercase							,     __lowercase       ):
																		raise ValueError(
																		    F'Unable to interleave a {dataset_type.__name__} (at position 0) with a {other_type.__name__} (at position {i}). Expected a list of Dataset objects or a list of IterableDataset objects.'       )
						if dataset_type is Dataset:
												return _concatenate_map_style_datasets(__lowercase							,     info=__lowercase							,     split=__lowercase							,     axis=__lowercase       )
						else:
												return _concatenate_iterable_datasets(__lowercase							,     info=__lowercase							,     split=__lowercase							,     axis=__lowercase       )




def 	snake_case__       (					__lowercase							,     __lowercase							,     __lowercase = 0							,     __lowercase = 0       )		->		int:







						"""simple docstring"""

						A__	:							Tuple						    = right or len(__lowercase       ) - 1
						if left > right:
												return -1
						elif list_data[left] == key:
												return left
						elif list_data[right] == key:
												return right
						else:
												return search(__lowercase							,     __lowercase							,     left + 1							,     right - 1       )


if __name__ == "__main__":
		import doctest

		doctest.testmod()