code: def test_distill_checkpointing_with_teacher(self):
      updates = dict(
          student_encoder_layers=2,
          student_decoder_layers=1,
          max_epochs=4,
          val_check_interval=0.25,
          alpha_hid=2.0,
          model_name_or_path="IGNORE_THIS_IT_DOESNT_GET_USED",
      )
      model = self._test_distiller_cli(updates, check_contents=False)
  
      ckpts = list(Path(model.output_dir).glob("*.ckpt"))
      self.assertEqual(1, len(ckpts))
      transformer_ckpts = list(Path(model.output_dir).glob("**/*.bin"))
      self.assertEqual(len(transformer_ckpts), 2)
      examples = lmap(
          str.strip,
          Path(model.hparams.data_dir)
          .joinpath("test.source")
          .open()
          .readlines(),
      )
      out_path = tempfile.mktemp()  # XXX: not being cleaned up
      generate_summaries_or_translations(
          examples, out_path, str(model.output_dir / "best_tfmr")
      )
      self.assertTrue(Path(out_path).exists())
  
      out_path_new = self.get_auto_remove_tmp_dir()
      convert_pl_to_hf(
          ckpts[0], transformer_ckpts[0].parent, out_path_new
      )
      assert os.path.exists(
          os.path.join(out_path_new, "pytorch_model.bin")
      )
  
imports: import argparse
  import logging
  import os
  import sys
  import tempfile
  from pathlib import Path
  import pytest
  import pytorch_lightning
  import torch
  from torch import nn
  import lightning_base
  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 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 4 for calling the TestSummarizationDistiller obj's underlying member method code with example usage: obj.test_distill_checkpointing_with_teacher() without return types