code: def make_test_data_dir(tmp_dir):
      for split in ["train", "val", "test"]:
          _dump_articles(
              os.path.join(tmp_dir, f"{split}.source"), ARTICLES
          )
          _dump_articles(
              os.path.join(tmp_dir, f"{split}.target"), SUMMARIES
          )
      return tmp_dir
  
imports: import argparse
  import logging
  import os
  import sys
  import tempfile
  from pathlib import Path
  import lightning_base
  import pytest
  import pytorch_lightning
  import torch
  from convert_pl_checkpoint_to_hf import convert_pl_to_hf
  from distillation import distill_main
  from finetune import SummarizationModule, main
  from huggingface_hub import list_models
  from parameterized import parameterized
  from run_eval import generate_summaries_or_translations
  from torch import nn
  from transformers import AutoConfig, AutoModelForSeq2SeqLM
  from transformers.testing_utils import CaptureStderr, CaptureStdout, TestCasePlus, require_torch_gpu, slow
  from utils import label_smoothed_nll_loss, lmap, load_json
label_desc: Use image node_id 2 for calling a global function with example usage: make_test_data_dir(tmp_dir) and returns: tmp_dir