def get_dataset_test_config():
  """Returns the test config for dataset."""
  batch_size = 4
  grid_size = 4
  img_dim = 15
  config = compositional.get_dataset_config()
  data_dir = os.path.join(
      FLAGS.test_srcdir,
      'language/gscan/xattn_model/testdata')
  config.data_dir = data_dir
  config.grid_size = grid_size
  config.train_split = 'train'
  config.eval_split = 'train'
  config.test_splits = ['train']
  config.split = 'test'
  config.num_epochs = 1
  config.img_dim = img_dim
  config.train_per_device_batch_size = batch_size
  config.eval_per_device_batch_size = batch_size

  return config
Returns the test config for dataset.
language/gscan/xattn_model/test_utils.py