





import itertools
import math






def 		lowerCAmelCase  (	UpperCamelCase__   :   int					)      ->    bool:

						"""simple docstring"""





						if 1 < number < 4:
												# 2 and 3 are primes
												return True
						elif number < 2 or number % 2 == 0 or number % 3 == 0:
												# Negatives, 0, 1, all even numbers, all multiples of 3 are not primes
												return False

						# All primes number are in format of 6k +/- 1
						for i in range(5       ,			int(math.sqrt(UpperCamelCase__					) + 1					)       ,			6					):
												if number % i == 0 or number % (i + 2) == 0:
																		return False
						return True






def 		lowerCAmelCase  (	)      ->    Any:

						"""simple docstring"""





						__SCREAMING_SNAKE_CASE:				List[str]           =   2
						while True:
												if is_prime(UpperCamelCase__					):
																		yield num
												num += 1






def 		lowerCAmelCase  (	UpperCamelCase__   :   int = 10_001					)      ->    int:

						"""simple docstring"""





						return next(itertools.islice(prime_generator()       ,			nth - 1       ,			UpperCamelCase__					)					)


if __name__ == "__main__":
							print(f'''{solution() = }''')










import unittest
from dataclasses import dataclass

import pytest

from accelerate.commands.config.config_args import SageMakerConfig
from accelerate.utils import ComputeEnvironment
from accelerate.utils.launch import _convert_nargs_to_dict





@dataclass
class 				a			(      __lowercase			):
				SCREAMING_SNAKE_CASE__     :   Any		      =     ComputeEnvironment.AMAZON_SAGEMAKER
				SCREAMING_SNAKE_CASE__     :   Optional[int]		      =     True
				SCREAMING_SNAKE_CASE__     :   int		      =     '''ml.p3.2xlarge'''
				SCREAMING_SNAKE_CASE__     :   Union[str, Any]		      =     '''accelerate_sagemaker_execution_role'''
				SCREAMING_SNAKE_CASE__     :   List[Any]		      =     '''hf-sm'''
				SCREAMING_SNAKE_CASE__     :   Any		      =     '''us-east-1'''
				SCREAMING_SNAKE_CASE__     :   Any		      =     1
				SCREAMING_SNAKE_CASE__     :   Dict		      =     '''accelerate-sagemaker-1'''
				SCREAMING_SNAKE_CASE__     :   Dict		      =     '''1.6'''
				SCREAMING_SNAKE_CASE__     :   int		      =     '''4.4'''
				SCREAMING_SNAKE_CASE__     :   int		      =     '''train.py'''
				SCREAMING_SNAKE_CASE__     :   Tuple		      =     [
				    '''--model_name_or_path''',
				    '''bert''',
				    '''--do_train''',
				    '''False''',
				    '''--epochs''',
				    '''3''',
				    '''--learning_rate''',
				    '''5e-5''',
				    '''--max_steps''',
				    '''50.5''',
				]
				SCREAMING_SNAKE_CASE__     :   List[str]		      =     [
				    '''--model_name_or_path''',
				    '''bert''',
				    '''--do_train''',
				    '''--do_test''',
				    '''False''',
				    '''--do_predict''',
				    '''--epochs''',
				    '''3''',
				    '''--learning_rate''',
				    '''5e-5''',
				    '''--max_steps''',
				    '''50.5''',
				]





class 				a			(      unittest.TestCase			):






				def 			snake_case_    (  self					):

										"""simple docstring"""




										__SCREAMING_SNAKE_CASE:				Optional[Any]           =   _convert_nargs_to_dict(MockLaunchConfig.success_training_script_args					)
										assert isinstance(converted_args['''model_name_or_path''']			, _lowerCAmelCase					)
										assert isinstance(converted_args['''do_train''']			, _lowerCAmelCase					)
										assert isinstance(converted_args['''epochs''']			, _lowerCAmelCase					)
										assert isinstance(converted_args['''learning_rate''']			, _lowerCAmelCase					)
										assert isinstance(converted_args['''max_steps''']			, _lowerCAmelCase					)

										with pytest.raises(_lowerCAmelCase					):
																_convert_nargs_to_dict(MockLaunchConfig.fail_training_script_args					)



