code: def test_terraformer_wmt_ende(self):
      batch_size_per_device = 2
      steps = 1
      n_layers = 2
      d_ff = 32
  
      gin.parse_config_file("terraformer_wmt_ende.gin")
  
      gin.bind_parameter("data_streams.data_dir", _TESTDATA)
      gin.bind_parameter(
          "batcher.batch_size_per_device", batch_size_per_device
      )
      gin.bind_parameter(
          "batcher.buckets",
          ([512], [batch_size_per_device, batch_size_per_device]),
      )
      gin.bind_parameter("train.steps", steps)
      gin.bind_parameter(
          "ConfigurableTerraformer.n_encoder_layers", n_layers
      )
      gin.bind_parameter(
          "ConfigurableTerraformer.n_decoder_layers", n_layers
      )
      gin.bind_parameter("ConfigurableTerraformer.d_ff", d_ff)
  
      output_dir = self.create_tempdir().full_path
      _ = trainer_lib.train(output_dir=output_dir)
  
file_path: trax/trax/models/research/terraformer_e2e_test.py
imports: import os
  from absl.testing import absltest
  import gin
  from trax import test_utils
  from trax.models.research import terraformer
  from trax.supervised import trainer_lib
  from trax.tf_numpy import numpy
label_desc: Use image node_id 2 for calling the TerraformerE2ETest obj's underlying member method code with example usage: obj.test_terraformer_wmt_ende() without return types