





'''simple docstring'''




import pytest

from datasets.parallel import ParallelBackendConfig, parallel_backend
from datasets.utils.py_utils import map_nested

from .utils import require_dill_gt_0_3_2, require_joblibspark, require_not_windows



def 		__A		(  lowerCamelCase_	):  # picklable for multiprocessing


		"""simple docstring"""



		return i + 1



@require_dill_gt_0_3_2
@require_joblibspark
@require_not_windows
def 		__A		(  ):


		"""simple docstring"""



		with parallel_backend("""spark"""	):
				assert ParallelBackendConfig.backend_name == "spark"

		SCREAMING_SNAKE_CASE :				List[Any]		=			[1, 2, 3]
		with pytest.raises(lowerCamelCase_	):
				with parallel_backend("""unsupported backend"""	):
						map_nested(lowerCamelCase_			,			lowerCamelCase_			,			num_proc=2	)

		with pytest.raises(lowerCamelCase_	):
				with parallel_backend("""unsupported backend"""	):
						map_nested(lowerCamelCase_			,			lowerCamelCase_			,			num_proc=-1	)



@require_dill_gt_0_3_2
@require_joblibspark
@require_not_windows
@pytest.mark.parametrize("""num_proc"""			,			[2, -1]	)
def 		__A		(  lowerCamelCase_	):


		"""simple docstring"""



		SCREAMING_SNAKE_CASE :				List[str]		=			[1, 2]
		SCREAMING_SNAKE_CASE :				Tuple		=			{"""a""": 1, """b""": 2}
		SCREAMING_SNAKE_CASE :				Union[str, Any]		=			{"""a""": [1, 2], """b""": [3, 4]}
		SCREAMING_SNAKE_CASE :				Dict		=			{"""a""": {"""1""": 1}, """b""": 2}
		SCREAMING_SNAKE_CASE :				Union[str, Any]		=			{"""a""": 1, """b""": 2, """c""": 3, """d""": 4}
		SCREAMING_SNAKE_CASE :				int		=			[2, 3]
		SCREAMING_SNAKE_CASE :				Union[str, Any]		=			{"""a""": 2, """b""": 3}
		SCREAMING_SNAKE_CASE :				Tuple		=			{"""a""": [2, 3], """b""": [4, 5]}
		SCREAMING_SNAKE_CASE :				Optional[int]		=			{"""a""": {"""1""": 2}, """b""": 3}
		SCREAMING_SNAKE_CASE :				Optional[int]		=			{"""a""": 2, """b""": 3, """c""": 4, """d""": 5}

		with parallel_backend("""spark"""	):
				assert map_nested(lowerCamelCase_			,			lowerCamelCase_			,			num_proc=lowerCamelCase_	) == expected_map_nested_sa
				assert map_nested(lowerCamelCase_			,			lowerCamelCase_			,			num_proc=lowerCamelCase_	) == expected_map_nested_sa
				assert map_nested(lowerCamelCase_			,			lowerCamelCase_			,			num_proc=lowerCamelCase_	) == expected_map_nested_sa
				assert map_nested(lowerCamelCase_			,			lowerCamelCase_			,			num_proc=lowerCamelCase_	) == expected_map_nested_sa
				assert map_nested(lowerCamelCase_			,			lowerCamelCase_			,			num_proc=lowerCamelCase_	) == expected_map_nested_sa










'''simple docstring'''




from ..utils import DummyObject, requires_backends




class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			Optional[int]      ,						*lowerCamelCase_				:			Any      ,						**lowerCamelCase_				:			Dict						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			Optional[int]      ,						*lowerCamelCase_				:			List[Any]      ,						**lowerCamelCase_				:			str						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			int      ,						*lowerCamelCase_				:			Union[str, Any]      ,						**lowerCamelCase_				:			int						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			List[str]      ,						*lowerCamelCase_				:			Union[str, Any]      ,						**lowerCamelCase_				:			str						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			Dict      ,						*lowerCamelCase_				:			Optional[Any]      ,						**lowerCamelCase_				:			Union[str, Any]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			List[Any]      ,						*lowerCamelCase_				:			Dict      ,						**lowerCamelCase_				:			Dict						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			Optional[Any]      ,						*lowerCamelCase_				:			Optional[int]      ,						**lowerCamelCase_				:			Union[str, Any]						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			str      ,						*lowerCamelCase_				:			int      ,						**lowerCamelCase_				:			Dict						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			Optional[Any]      ,						*lowerCamelCase_				:			str      ,						**lowerCamelCase_				:			List[Any]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			List[Any]      ,						*lowerCamelCase_				:			Tuple      ,						**lowerCamelCase_				:			int						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			int      ,						*lowerCamelCase_				:			Dict      ,						**lowerCamelCase_				:			Tuple						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			Any      ,						*lowerCamelCase_				:			str      ,						**lowerCamelCase_				:			Optional[Any]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			List[str]      ,						*lowerCamelCase_				:			Dict      ,						**lowerCamelCase_				:			int						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			Any      ,						*lowerCamelCase_				:			Any      ,						**lowerCamelCase_				:			Any						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			Any      ,						*lowerCamelCase_				:			Optional[int]      ,						**lowerCamelCase_				:			str						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			Any      ,						*lowerCamelCase_				:			List[Any]      ,						**lowerCamelCase_				:			Any						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			Union[str, Any]      ,						*lowerCamelCase_				:			Union[str, Any]      ,						**lowerCamelCase_				:			Union[str, Any]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			List[Any]      ,						*lowerCamelCase_				:			str      ,						**lowerCamelCase_				:			Optional[int]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			Dict      ,						*lowerCamelCase_				:			Optional[Any]      ,						**lowerCamelCase_				:			int						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			int      ,						*lowerCamelCase_				:			Tuple      ,						**lowerCamelCase_				:			Dict						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			Tuple      ,						*lowerCamelCase_				:			int      ,						**lowerCamelCase_				:			Union[str, Any]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			Optional[Any]      ,						*lowerCamelCase_				:			Dict      ,						**lowerCamelCase_				:			List[Any]						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			List[Any]      ,						*lowerCamelCase_				:			Optional[Any]      ,						**lowerCamelCase_				:			List[str]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			List[str]      ,						*lowerCamelCase_				:			Tuple      ,						**lowerCamelCase_				:			Tuple						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			Dict      ,						*lowerCamelCase_				:			int      ,						**lowerCamelCase_				:			Any						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			int      ,						*lowerCamelCase_				:			int      ,						**lowerCamelCase_				:			Any						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			List[Any]      ,						*lowerCamelCase_				:			Tuple      ,						**lowerCamelCase_				:			Tuple						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			int      ,						*lowerCamelCase_				:			Tuple      ,						**lowerCamelCase_				:			Any						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			List[Any]      ,						*lowerCamelCase_				:			Tuple      ,						**lowerCamelCase_				:			Optional[Any]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			List[Any]      ,						*lowerCamelCase_				:			Optional[int]      ,						**lowerCamelCase_				:			Any						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			List[Any]      ,						*lowerCamelCase_				:			List[Any]      ,						**lowerCamelCase_				:			List[Any]						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			List[Any]      ,						*lowerCamelCase_				:			Optional[Any]      ,						**lowerCamelCase_				:			Optional[int]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			Any      ,						*lowerCamelCase_				:			Dict      ,						**lowerCamelCase_				:			Optional[int]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			Any      ,						*lowerCamelCase_				:			Dict      ,						**lowerCamelCase_				:			int						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			List[Any]      ,						*lowerCamelCase_				:			Union[str, Any]      ,						**lowerCamelCase_				:			List[str]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			Optional[Any]      ,						*lowerCamelCase_				:			Dict      ,						**lowerCamelCase_				:			Optional[Any]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)







class      UpperCamelCase__					(					metaclass=lowercase_			):




					"""simple docstring"""
					SCREAMING_SNAKE_CASE__				=	['''flax''']


					def __init__(						self				:			Tuple      ,						*lowerCamelCase_				:			Tuple      ,						**lowerCamelCase_				:			Union[str, Any]						):

							'''simple docstring'''



							requires_backends(self      ,						["""flax"""]						)


					@classmethod
					def lowerCamelCase_			(						cls				:			Optional[Any]      ,						*lowerCamelCase_				:			Any      ,						**lowerCamelCase_				:			Optional[int]						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



					@classmethod
					def lowerCamelCase_			(						cls				:			List[Any]      ,						*lowerCamelCase_				:			Any      ,						**lowerCamelCase_				:			Tuple						):

							'''simple docstring'''



							requires_backends(cls      ,						["""flax"""]						)



