code: 
  
  
  
  
  """simple docstring"""
  
  
  
  # as due to their complexity multi-gpu tests could impact other tests, and to aid debug we have those in a separate module.
  
  import os
  import sys
  
  from transformers.testing_utils import TestCasePlus, execute_subprocess_async, get_gpu_count, require_torch_gpu, slow
  
  from .utils import load_json
  
  
  
  
  class 			UpperCAmelCase_						(		A_							):
  
  
  
  
      @classmethod
      def 							__magic_name__							(					cls		: List[str]   )   ->							Optional[int]:
  
  
  
  
       '''simple docstring'''
  
       return cls
  
  
  
  
      @slow
      @require_torch_gpu
      def 							__magic_name__							(					self		: int   )   ->							Dict:
  
  
  
  
       '''simple docstring'''
  
       A__ =       self.get_auto_remove_tmp_dir()
       A__ =       F"""
              --model_name Helsinki-NLP/opus-mt-en-ro
              --save_dir {output_dir}
              --data_dir {self.test_file_dir_str}/test_data/wmt_en_ro
              --num_beams 2
              --task translation
          """.split()
  
       # we want this test to run even if there is only one GPU, but if there are more we use them all
       A__ =       get_gpu_count()
       A__ =       F"""
              -m torch.distributed.launch
              --nproc_per_node={n_gpu}
              {self.test_file_dir}/run_distributed_eval.py
          """.split()
       A__ =       [sys.executable] + distributed_args + args
       execute_subprocess_async(snake_case_			,				env=self.get_env()   )
  
       A__ =       os.path.join(snake_case_			,				"test_bleu.json"   )
       A__ =       load_json(snake_case_   )
       # print(metrics)
       self.assertGreaterEqual(metrics["bleu"]			,				25   )
  
  
  
  
  
  
style_context: 
  
  
  
  
  """simple docstring"""
  
  
  
  from transformers import PretrainedConfig
  
  
  
  
  class 			UpperCAmelCase_						(		A_							):
      lowercase__		      =   '''custom'''
  
  
  
  
      def __init__(					self		: Any			,				snake_case_		: List[Any]=1			,				**snake_case_		: Any   )   ->							str:
  
  
  
  
       '''simple docstring'''
  
       A__ =       attribute
       super().__init__(**snake_case_   )
  
  
  
  
  
  
  class 			UpperCAmelCase_						(		A_							):
      lowercase__		      =   '''custom'''
  
  
  
  
      def __init__(					self		: Any			,				snake_case_		: Tuple=1			,				**snake_case_		: Dict   )   ->							Any:
  
  
  
  
       '''simple docstring'''
  
       A__ =       attribute
  
  
  
  
  
  