"""simple docstring"""


import argparse

import torch

from transformers import MobileBertConfig, MobileBertForPreTraining, load_tf_weights_in_mobilebert
from transformers.utils import logging


logging.set_verbosity_info()







def 					__a   (		A    ,				A    ,				A      ):






     '''simple docstring'''



     lowercase__				 =							MobileBertConfig.from_json_file(A      )
     print(f'''Building PyTorch model from configuration: {config}'''      )
     lowercase__				 =							MobileBertForPreTraining(A      )
     # Load weights from tf checkpoint
     lowercase__				 =							load_tf_weights_in_mobilebert(A    ,				A    ,				A      )
     # Save pytorch-model
     print(f'''Save PyTorch model to {pytorch_dump_path}'''      )
     torch.save(model.state_dict()    ,				A      )


if __name__ == "__main__":
 lowerCAmelCase_:					Optional[Any]     =					argparse.ArgumentParser()
 # Required parameters
 parser.add_argument(
     "--tf_checkpoint_path", default=None, type=str, required=True, help="Path to the TensorFlow checkpoint path."
 )
 parser.add_argument(
     "--mobilebert_config_file",
     default=None,
     type=str,
     required=True,
     help=(
         "The config json file corresponding to the pre-trained MobileBERT model. \n"
         "This specifies the model architecture."
     ),
 )
 parser.add_argument(
     "--pytorch_dump_path", default=None, type=str, required=True, help="Path to the output PyTorch model."
 )
 lowerCAmelCase_:					Optional[Any]     =					parser.parse_args()
 convert_tf_checkpoint_to_pytorch(args.tf_checkpoint_path, args.mobilebert_config_file, args.pytorch_dump_path)





"""simple docstring"""


from __future__ import annotations

import math







def 					__a   (		A      ):






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


lowerCAmelCase_:					Optional[Any]     =					[num for num in range(3, 1_0_0_0_0_1, 2) if not is_prime(num)]







def 					__a   (		A      ):






     '''simple docstring'''



     if not isinstance(A    ,				A      ):
          raise ValueError("n must be an integer"      )
     if n <= 0:
          raise ValueError("n must be >= 0"      )

     lowercase__				 =							[]
     for num in range(len(A      )      ):
          lowercase__				 =							0
          while 2 * i * i <= odd_composites[num]:
               lowercase__				 =							odd_composites[num] - 2 * i * i
               if is_prime(A      ):
                    break
               i += 1
          else:
               list_nums.append(odd_composites[num]      )
               if len(A      ) == n:
                    return list_nums

     return []







def 					__a   (		):






     '''simple docstring'''



     return compute_nums(1      )[0]


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




