

from math import sqrt





def    __lowerCamelCase      (					UpperCamelCase__ ):







							'''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(sqrt(UpperCamelCase__ ) + 1 )  ,				6 ):
														if number % i == 0 or number % (i + 2) == 0:
																					return False
							return True





def    __lowerCamelCase      (					UpperCamelCase__ = 10001 ):







							'''simple docstring'''






							snake_case_     							=       0
							snake_case_     							=       1
							while count != nth and number < 3:
														number += 1
														if is_prime(UpperCamelCase__ ):
																					count += 1
							while count != nth:
														number += 2
														if is_prime(UpperCamelCase__ ):
																					count += 1
							return number


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







import tempfile
import unittest

from make_student import create_student_by_copying_alternating_layers

from transformers import AutoConfig
from transformers.file_utils import cached_property
from transformers.testing_utils import require_torch


_UpperCAmelCase						:  Dict									=			"""sshleifer/bart-tiny-random"""
_UpperCAmelCase						:  Optional[Any]									=			"""patrickvonplaten/t5-tiny-random"""







@require_torch
class 					lowercase							(			unittest.TestCase     ):




				@cached_property
				def        a				(   self			):
											return AutoConfig.from_pretrained(snake_case			)




				def        a				(   self			):
											snake_case_							,       *snake_case_     							=       create_student_by_copying_alternating_layers(snake_case ,  tempfile.mkdtemp() ,  e=1 ,  d=1			)
											self.assertEqual(student.config.num_hidden_layers ,  1			)




				def        a				(   self			):
											snake_case_							,       *snake_case_     							=       create_student_by_copying_alternating_layers(snake_case ,  tempfile.mkdtemp() ,  e=1 ,  d=snake_case			)




				def        a				(   self			):
											snake_case_							,       *snake_case_     							=       create_student_by_copying_alternating_layers(snake_case ,  tempfile.mkdtemp() ,  e=1 ,  d=snake_case			)
											self.assertEqual(student.config.encoder_layers ,  1			)
											self.assertEqual(student.config.decoder_layers ,  self.teacher_config.encoder_layers			)




				def        a				(   self			):
											snake_case_							,       *snake_case_     							=       create_student_by_copying_alternating_layers(snake_case ,  tempfile.mkdtemp() ,  e=1 ,  d=1			)
											self.assertEqual(student.config.encoder_layers ,  1			)
											self.assertEqual(student.config.decoder_layers ,  1			)




				def        a				(   self			):
											with self.assertRaises(snake_case			):
																		create_student_by_copying_alternating_layers(snake_case ,  tempfile.mkdtemp() ,  e=snake_case ,  d=snake_case			)




